TorchDynamo 源码剖析 01 - Frame Evaluation 与字节码基础

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.

简介 Link to heading PyTorch 最大的优点是其灵活性，但缺点也很显著，PyTorch 生态对图编译器 (graph compiler) 的支持非常有限，现有的基于 TorchScript 的方案非常难用。为了解决这些问题，PyTorch 2.0 引入了 torch.compile ，它包含几个新的组件: TorchDynamo, AOTAutograd, PrimTorch, TorchInductor。其中，TorchDynamo 用于从用户的 PyTorch 代码中以最小的代价 捕获计算图 ，更多关于 PyTorch 2.0 的简介请参考 PyTorch 2.0 。

Read on fkong.tech ↗