写点什么

TorchDynamo

0 人感兴趣 · 1 次引用

  • 最新
  • 推荐
https://static001.geekbang.org/infoq/01/01cd3f0378fa8386784644d3ed17bf2b.jpeg?x-oss-process=image%2Fresize%2Cw_416%2Ch_234

TorchDynamo 初探:Python ByteCode 的动态修改

用户头像
OneFlow
2023-01-03

深度学习框架编译优化时,需要先根据计算逻辑形成一个逻辑计算图,然后再改写计算图,最后执行改写后的计算图。其中生成逻辑计算图方式有两种。

TorchDynamo_TorchDynamo技术文章_InfoQ写作社区