![](https://img-blog.csdnimg.cn/20201014180756913.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
pytorch源码
文章平均质量分 96
哦豁灬
这个作者很懒,什么都没留下…
展开
-
torch._dynamo 理解(2)——Backend
TorchDynamo 是一个 Python 级别的即时 (JIT) 编译器,旨在让未修改的 PyTorch 程序运行得更快。它通过 Python Frame Evaluation Hooks(Python 框架评估钩子)来实现这一目标,以便在运行时动态地生成和优化代码。这使得 TorchDynamo 可以有效地处理各种 Python 代码,包括包含控制流(如循环和条件语句)的代码,而无需进行任何修改。原创 2024-07-30 13:40:15 · 543 阅读 · 0 评论 -
torch._dynamo 理解(1)——Frontend
TorchDynamo 是一个 Python 级别的即时 (JIT) 编译器,旨在让未修改的 PyTorch 程序运行得更快。它通过 Python Frame Evaluation Hooks(Python 框架评估钩子)来实现这一目标,以便在运行时动态地生成和优化代码。这使得 TorchDynamo 可以有效地处理各种 Python 代码,包括包含控制流(如循环和条件语句)的代码,而无需进行任何修改。前端的整体流程如下:nn.ReLU(),).cuda()model(x)sub = None。原创 2024-07-30 13:38:28 · 943 阅读 · 0 评论