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

PyTorch 接入 FlyDSL 后端加速 AMD MI350 GPU 算子

1 篇报道1 个报道来源3 小时前更新

先了解这件事

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 根据报道生成 · 2 小时前更新

报道时间线

沿着报道,了解事件的不同侧面。

10月22日
  1. PyTorch Blog
    PyTorch 借助 FlyDSL 在 AMD GPU 上实现加速

    PyTorch 用户现可在 AMD MI350 系列 GPU 上通过现有算子 API 使用 FlyDSL 后端,覆盖 dense/grouped GEMM、FMHA 注意力、MXFP8/MXFP4 scaled GEMM、RMSNorm 和 TopK,无需调用自定义 kernel。

本事件热度走势

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