fx-graph

一篇博文，解释 PyTorch FX 图，这是 PyTorch 2.0 编译生态系统中使用的一种中间表示。它涵盖了核心对象 Graph、Node 和 GraphModule，以及如何理解和使用它们。

本文解释了 PyTorch 中的 torch.compile 堆栈，详述了从 API 到 Dynamo、FX 图、ATen 操作以及用于 JIT 编译的 Torch Inductor 的各个步骤。