投稿

PyTorch Dispatcher原理！PyTorch算子注册！【AI编译器之PyTorch】系列06篇

2811

1

2023-01-15 23:20:39

【AI编译器之PyTorch】系列06篇：PyTorch Dispatcher原理！PyTorch算子注册！—— PyTorch2.0出来之后，AOTAutograd能够获取反向图，而起主要是通过Pytorch Dispatcher机制来获取Autograd的结果转换成FX IR。Pytorch Dispatcher可以理解为分发器，可以根据关于 tensor 输入的一些信息来决定要调用哪一块的程序，下面来展开Dispatcher原理，PyTorch算子注册方式！

深度学习编译器

传统编译器

ZOMI酱发消息

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

回归搜索的本质！没有广告，直达结果！

【AI编译器】PyTorch编译优化 (6/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

顶部