
关注程旭源,我们一起向上生长


本文是为了让大家快速入门和使用Torch 2.x,所以主要结合代码去介绍如何使用TorchDynamo和TorchInductor,并探索其原理。

Pytorch官方团队宣布Pyotch 2.X的新时代已经到来。
主要四个方面:
-
TorchDynamo:把Python Bytecode 字节码分析生成 FX 图;
-
TorchInductor:神经网络编译器;
-
AOTAutograd :以 ahead-of-time 的方式生成反向图;
-
PrimTorch :引入一个小型算子集,使后端更容易。
原创文章。转载请注明:
作者:meixi
网址: https://www.icnma.com