随笔分类 -  Machine Learning

PyTorch JIT编译器
摘要：PyTroch似乎引进了一些有趣的JIT编译功能，用来优化性能，包括做一些fusion。 可以看下的项目： torchdynymo: https://github.com/pytorch/torchdynamo 似乎用了triton NVFuser: https://github.com/pytor 阅读全文
posted @ 2022-07-19 14:23 xuyv 编辑

点击右上角即可分享
微信分享提示