AMD工程师利用AI帮助制作纯Python AMD GPU用户空间驱动程序
内容摘要
AMD人工智能软件副总裁Anush Elangovan利用Claude Code（一个AI代理）开发了一个纯Python的AMD GPU用户空间驱动程序。该驱动程序绕过了标准的ROCm/HIP用户空间堆栈，通过ctypes ioctl直接与/dev/kfd和/dev/dri/renderD*通信，主要目的是用于测试其他ROCm代码和协助调试用户空间堆栈，特别是SDMA和计算/通信重叠方面。
该驱动程序受到Tinygrad实现的启发，支持KFD ioctl绑定、GPU系列注册（RDNA2/3/4, CDNA2/3）、SDMA复制引擎、PM4计算包构建和时间线信号量等功能。Elangovan指出，AI代理在软件开发速度上是“伟大的均衡器”。该项目已迅速扩展，加入了多GPU支持和计算密集型内核功能，初始提交显示在MI300X/gfx942硬件上通过了130个测试。
(来源：Phoronix)