其他分享
首页 > 其他分享> > PyTorch JIT编译器

PyTorch JIT编译器

作者:互联网

PyTroch似乎引进了一些有趣的JIT编译功能,用来优化性能,包括做一些fusion。

可以看下的项目:

torchdynymo: https://github.com/pytorch/torchdynamo 

似乎用了triton

NVFuser: https://github.com/pytorch/pytorch/blob/release/1.12/torch/csrc/jit/codegen/cuda/README.md

NVFuser is integrated as a backend for TorchScript's Profiling Graph Executor. NVFuser is the default fuser for NVIDIA GPUs.

 

 

标签:github,JIT,https,编译器,pytorch,NVFuser,PyTorch,com
来源: https://www.cnblogs.com/simpleminds/p/16493917.html