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·原文发布· 3 小时前AI 评分42
PyTorch 借助 FlyDSL 在 AMD GPU 上实现加速
Accelerating PyTorch on AMD GPUs with FlyDSL
AI 导读
PyTorch 用户现可在 AMD MI350 系列 GPU 上通过现有算子 API 使用 FlyDSL 后端，覆盖 dense/grouped GEMM、FMHA 注意力、MXFP8/MXFP4 scaled GEMM、RMSNorm 和 TopK，无需调用自定义 kernel。
本站提供 AI 导读，完整报道请前往来源网站阅读。
阅读原文 PyTorch Blog · pytorch.org