🟢 🔧 硬件 发布于: · 1 分钟阅读 ·

AMD: FlyDSL——Python DSL 承诺以更少代码实现手写 HIP C++ GPU 核函数的性能

编辑插图:Python 蛇缠绕 GPU 芯片,芯片中流出优化的数据流

FlyDSL 是 AMD 的 Python 领域特定语言,用于编写 GPU 核函数,根据新的 ROCm 指南,其性能可媲美手动优化的 HIP C++ 代码,同时代码量更少。AMD 发布了移植现有 HIP 核函数的实用指南,将 FlyDSL 定位为 NVIDIA CUDA 工具链的 Python 优先替代方案。

🤖

本文由人工智能基于一手来源生成。

AMD 于 2026 年 7 月 9 日在 ROCm 博客上发布了将 HIP 核函数移植到 FlyDSL 的实用指南,FlyDSL 是其 GPU 编程 Python 领域特定语言。HIP(Heterogeneous-compute Interface for Portability)是 AMD 的 C++ API——对应 NVIDIA 的 CUDA 平台——而 FlyDSL 将同等级别的控制力迁移到 Python。

为什么要为 GPU 核函数使用 Python?

如今进入 GPU 优化的门槛很高:手写 C++ 核函数需要只有少数工程师才具备的专业知识。FlyDSL 承诺以大幅减少的代码量实现手动优化的 HIP 代码性能——这与 OpenAI 的 Triton 赢得 NVIDIA 侧 ML 社区的论点相同,PyTorch 本周还将其扩展了 CuTeDSL 后端。AMD 的指南逐步演示了现有高性能核函数的转换,表明公司的目标是已有 HIP 代码但希望加快迭代速度的团队。

战略背景

CUDA 生态系统依然是 NVIDIA 最深的护城河:竞争对手的硬件往往具有可比性,但软件工具链不然。因此 FlyDSL 不仅仅是一个开发者小工具——Python 优先层降低了迁移到 AMD Instinct 加速器的成本,客户不仅需要被 MI 硬件说服,还需要相信他们的工程师能在上面编写高性能代码。

缺少什么

已发布的指南没有提供 FlyDSL 与具体核函数上手写 HIP 的对比基准数字,因此性能相当的说法还需独立验证。真正的考验将是 vLLM 等开源项目的采用情况——这些项目 AMD 后端此前一直依赖手写核函数。

常见问题

什么是 FlyDSL?
FlyDSL 是嵌入 Python 的 AMD 领域特定语言,用于在 ROCm 平台上编写 GPU 核函数——目标是以 Python 的简洁实现手动优化 HIP C++ 代码的性能。
什么是 HIP?
HIP(Heterogeneous-compute Interface for Portability)是 AMD 的 GPU 编程 C++ API,设计为 NVIDIA CUDA 平台的可移植对应物。