跳到主要内容
检索

# CUDA Tile IR

1 篇关联报道

订阅更新
  1. NVIDIA Technical Blog

    团队用多智能体技能把24个TileGym算子从cuTile Python/Triton-TileIR移植到cuTile Rust,平均达到参考性能的99.5%

    该工作构建了一个多智能体AI技能,将cuTile Python与Triton-TileIR编写的GPU tile内核翻译为cuTile Rust,并据此把全部24个公开TileGym算子(约40个内核,覆盖逐元素运算到flash-attention decode、MLA与MoE)移植完成,在NVIDIA DGX B200上以CUPTI设备时间测得几何平均0.995,即平均达到cuTile Python性能的99.5%,且每个阶段都以可机器校验的判定和Tile IR diff把关。

已读完全部报道