AMDエンジニアがAIを活用し、純粋なPythonによるAMD GPUユーザー空間ドライバーの作成を支援
概要
AMDのAIソフトウェア担当VPであるAnush Elangovan氏は、Claude Code（AIエージェント）を利用して、AMD GPU用の純粋なPythonによるユーザー空間ドライバーを作成しました。このドライバーは、標準の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)