IXThinker·2025-07-13 12:50

純Python寫內核 QuACK為H100提速33%

Flash Attention與Mamba作者之一Tri Dao再出新作：他與兩位普林斯頓計算機博士生提出內核庫QuACK，完全以Python編寫、零CUDA代碼，卻能在H100上比深度優化庫再快33%至50%。

純Python寫內核

QuACK是一個SOL內存綁定內核庫，藉助CuTe-DSL實現，全程沒有使用任何CUDA C++代碼。這意味著開發者毋須掌握CUDA，也能寫出高性能GPU內核，大幅降低調優門檻。