AMD 工程师利用人工智能帮助开发纯 Python AMD GPU 用户空间驱动程序

AMD 人工智能软件副总裁 Anush Elangovan 使用 Claude Code 帮助编写了一个纯 Python 的 AMD GPU 用户空间驱动程序。这个 Python 用户空间驱动程序目前正在创建中，旨在帮助测试其他 ROCm 代码，并在通过 ROCm/HIP 用户空间堆栈时进行调试。

Anush 受到 Tinygrad 用户空间 AMD GPU 驱动程序实现的启发，并使用 Claude AI 创建了一个用于 SDMA 和计算/通信重叠调试的压力测试用户空间驱动程序。Anush 在 X 上发帖：“我一次都没有打开过编辑器。[AI]代理是软件中的伟大平等主义者。速度是护城河。”

随着进一步的工作，该用户空间驱动程序也已开始支持计算受限的内核。

该用户空间驱动程序目前通过此 GitHub 分支进行开发。最初的提交说明了目前已就绪的首要功能：

“添加纯 Python AMD GPU 用户空间驱动程序
一个独立的 Python 驱动程序，通过 ctypes ioctls 直接与 /dev/kfd 和 /dev/dri/renderD* 通信，绕过 ROCm/HIP 用户空间堆栈。支持 KFD 后端，具有可插拔架构，以支持未来的裸机 PCI (AM) 后端。

特性：
– KFD ioctl 绑定（队列、内存、事件）
– GPU 系列注册表（RDNA2/3/4，CDNA2/3）
– SDMA 拷贝引擎，支持线性拷贝和栅栏包
– PM4 计算数据包构建器（调度、释放内存等）
– GPU-CPU 同步的时间线信号量
– /sys/devices/virtual/kfd/kfd 的拓扑解析器
– 内核加载的 ELF 代码对象解析器
– 130 项测试通过（MI300X/gfx942 上的单元测试+集成测试）

合著者：Claude (claude-opus-4-6)”

在过去的两天里，这个纯 Python AMD 用户空间驱动程序已经扩展以包括多 GPU 支持、计算受限的内核和其他功能。看到这个基于 Python 的 AMD GPU 用户空间驱动程序的工作将走向何方将很有趣。

转自  AMD Engineer Leverages AI To Help Make A Pure-Python AMD GPU User-Space Driver – Phoronix