产品发布 / 更新普通
PyTorch 借助 FlyDSL 加速 AMD GPU 上的算子性能
原始标题:Accelerating PyTorch on AMD GPUs with FlyDSL
内容摘要
PyTorch 在 AMD MI350 系列 GPU 上通过现有算子 API 接入 FlyDSL 后端,覆盖 dense/grouped GEMM、FMHA、MXFP8/MXFP4 缩放 GEMM、RMSNorm 和 TopK,无需调用自定义 kernel。
内容分类AI 产品发布与更新
内容层级普通情报
发布时间(北京时间)
本站收录时间(北京时间)
信息来源PyTorch:Blog(RSS)
站内情报编号intel-b58a6538a04286b3aff6ac6f