AMD在MI350上通过FlyDSL加速PyTorch算子

2026年10月22日，AMD与Meta团队在PyTorch博客发文介绍，AMD MI350系列GPU（gfx950架构）用户现可通过现有算子API使用FlyDSL后端，无需应用调用自定义内核。覆盖范围包括Dense与Grouped GEMM、融合多头注意力（FMHA/Attention）、MXFP8/MXFP4缩放GEMM、RMSNorm和TopK。使用条件方面，当前支持仅面向MI350系列gfx950 GPU，示例需gfx950 GPU，ROCm构建沿用PyTorch的“cuda”设备字符串；除注意力外的算子需从受支持的0.3.x系列安装可选运行时。注意力SDPA无需FlyDSL包或配置，只需PyTorch构建链接AOTriton 0.14.2b或更高版本，由后者内置预编译内核。这是该事件目前唯一的主报道，尚无跟进报道或与早先说法冲突之处。

AI 根据报道生成 · 3 小时前更新

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