PyTorch 接入 FlyDSL 加速 AMD MI350 GPU 算子
热点事件持续更新
PyTorch 接入 FlyDSL 加速 AMD MI350 GPU 算子
1 篇报道1 个报道来源3 小时前更新
先了解这件事
AI 综述
PyTorch 在 AMD MI350 系列 GPU 上通过现有算子 API 接入 FlyDSL 后端,覆盖 dense/grouped GEMM、融合多头注意力(FMHA)、MXFP8/MXFP4 缩放 GEMM、RMSNorm 和 TopK,应用无需调用自定义 kernel。 目前支持仅限采用 gfx950 架构的 MI350 系列 GPU。该后端为可选:GEMM 经 torch.compile 生效,符合条件的 RMSNorm 和 TopK 在启用后于 eager 模式下自动分派;不支持的输入仍走原有 PyTorch 实现。 PyTorch 称这一集成可加速上述算子,但报道未给出具体性能数据。
AI 根据报道生成 · 2 小时前更新
最新进展10月22日 01:45
PyTorch 借助 FlyDSL 加速 AMD GPU 上的算子性能报道时间线
沿着报道,了解事件的不同侧面。
10月22日
- PyTorch:BlogPyTorch 借助 FlyDSL 加速 AMD GPU 上的算子性能
PyTorch 在 AMD MI350 系列 GPU 上通过现有算子 API 接入 FlyDSL 后端,覆盖 dense/grouped GEMM、FMHA、MXFP8/MXFP4 缩放 GEMM、RMSNorm 和 TopK,无需调用自定义 kernel。
本事件热度走势
还没有足够的连续观测数据,暂不绘制趋势。