投稿

PyTorch静态图的TorchFX和Lazy Tensor特性解读！【AI编译器之PyTorch】系列03篇

3035

2

2022-12-11 12:31:05

【AI编译器之PyTorch】系列03篇：PyTorch静态图TorchFX和Lazy Tensor特性解读！PyTorch在静态图走过弯路有哪些？ PyTorch2.0出来之后，最重要就是加入了AI编译器，通过TorchDynamo特性把Torch的动态图转为静态图。但是PyTorch的静态图方案不不是一帆风顺，前后经历了Script、FX、Lazy Tensor等不同的过度方案。PyTorch在静态图走过的弯路有哪些？我们来细数PyTorch静态图FX和Lazy Tensor特性吧！

深度学习编译器

传统编译器

ZOMI酱发消息

昇腾招人，已经毕业的快联系鸭

【AI编译器】PyTorch编译优化 (3/6)

自动连播

3.4万播放

简介

PyTorch 2.0 是PyTorch捐献给Linux基金会的第一个版本。从PyTorch 1.0到1.13，PyTorch进行了很多创新和迭代。而2.0引入了torch.compile 编译模式，是PyTorch真正从学术界迈向工业界的很重要一步，未来将会如何发展？PyTorch又将会如何引领AI编译器领域呢？

01. PyTorch 2.0新特性

13:30

02. TorchScript 图模式第一次尝试

08:30

03. TorchFX和Lazy Tensor特性

13:41

04 TorchDynamo原理

14:45

05 AOTAutograd原理

10:42

06 PyTorch Dispatcher方式

15:49

顶部