PyTorch 接入 FlyDSL 后端加速 AMD MI350 GPU 算子
热点事件持续更新
PyTorch 接入 FlyDSL 后端加速 AMD MI350 GPU 算子
1 篇报道1 个报道来源2 小时前更新
先了解这件事
AI 综述
AMD 宣布 FlyDSL 作为可选后端接入 PyTorch:MI350 系列 GPU 用户通过现有算子 API 即可使用 dense/grouped GEMM、FMHA 注意力、MXFP8/MXFP4 scaled GEMM、RMSNorm 和 TopK,无需应用调用自定义 kernel。 目前支持仅限采用 gfx950 架构的 AMD MI350 系列 GPU,需使用较新的 ROCm PyTorch nightly 并安装 0.3.x 系列的可选运行时;不支持的输入仍走 PyTorch 原有实现。
AI 根据报道生成 · 1 小时前更新
最新进展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 scaled GEMM、RMSNorm 和 TopK,无需调用自定义 kernel。
本事件热度走势
还没有足够的连续观测数据,暂不绘制趋势。