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