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·· 15 天前AI 评分58
AMD 与 Meta 将 FlyDSL 接入 PyTorch，加速 MI350 系列 GPU 算子
Accelerating PyTorch on AMD GPUs with FlyDSL
AI 导读
AMD 与 Meta 作者介绍，PyTorch 用户可在 MI350 系列 GPU 上通过现有算子 API 使用 FlyDSL，覆盖 Dense/Grouped GEMM、MXFP8/MXFP4 缩放 GEMM、RMSNorm 与 TopK，注意力内核随 AOTriton 0.14b 系列提供。
来源：PyTorch：Blog · pytorch.org