正在加载陈远的个人网站

产品

PyTorch 借助 FlyDSL 加速 AMD GPU 上的算子性能

PyTorch:Blog(RSS)

导读

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