跳到正文
热点事件持续更新

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日
  1. PyTorch:Blog
    PyTorch 借助 FlyDSL 加速 AMD GPU 上的算子性能

    PyTorch 在 AMD MI350 系列 GPU 上通过现有算子 API 接入 FlyDSL 后端,覆盖 dense/grouped GEMM、FMHA、MXFP8/MXFP4 缩放 GEMM、RMSNorm 和 TopK,无需调用自定义 kernel。

本事件热度走势

还没有足够的连续观测数据,暂不绘制趋势。