AMD 与 Meta 将 FlyDSL 后端接入 PyTorch

AMD 与 Meta 在 PyTorch 博客宣布，FlyDSL 已作为可选后端接入 PyTorch，面向采用 gfx950 架构的 AMD MI350 系列 GPU。用户可通过现有算子 API 使用稠密与分组 GEMM、FMHA 注意力、MXFP8/MXFP4 缩放 GEMM、RMSNorm 和 TopK，无需调用自定义 kernel。 按官方说明，GEMM 经 torch.compile 使用该后端；在启用且可用时，符合条件的 RMSNorm 和 TopK 调用会在 eager 模式下自动分派；不支持的输入仍走原有 PyTorch 实现。FlyDSL 注意力 kernel 自 AOTriton 0.14b 系列起作为额外后端提供，AOTriton 是 PyTorch ROCm 的 F.scaled_dot_product_attention 所依赖的库。

AI 根据报道生成 · 33 分钟前更新

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