PyTorch Blog· AMD: Xiaobing Zhang, Yutao Xu, Rui Li, Heng Yi, Xinya Zhang, Shijie Feng, Felix Li, Jeff Daily, Jiong Gong, Hattie Wu, Peng Sun, Emad Barsoum; META: Jason Ansel and Simon Layton·· 2 小时前AI 评分53
PyTorch 在 AMD MI350 系列 GPU 上集成 FlyDSL 后端加速
Accelerating PyTorch on AMD GPUs with FlyDSL
阅读原文
本站未展示全文，请前往来源网站阅读。
AI 导读
PyTorch 宣布在 AMD MI350 系列（gfx950）GPU 上可选启用 FlyDSL 后端，覆盖稠密与分组 GEMM、MXFP8/MXFP4 scaled GEMM、RMSNorm、TopK，注意力内核则通过 AOTriton 0.14.2b 提供。
来源：PyTorch Blog · pytorch.org