torch.compile 重要步骤函数调用栈

This site does not allow itself to be embedded. You can still read it on the original site — the toolbar below keeps your place in the directory.

前段时间因为工作原因重新浏览了 torch.compile 中几个重要的函数调用栈，现将这些内容分享给有兴趣通过源代码理解 torch.compile 行为的朋友。 本文使用 PyTorch 官方提供的 docker 镜像: pytorch/pytorch:2.4.0-cuda12.4-cudnn9-devel ，PyTorch 版本为 2.4.0。

Read on fkong.tech ↗