NVIDIA Technical Blog 发布于 09/17 00:28

Translating CUDA Tile Operations from Python to Rust Using Agentic AI

(翻译)使用 Agentic AI 将 CUDA Tile 操作从 Python 翻译为 Rust

查看原文

资讯摘要

cuTile Rust () is a tile-based system for safe, idiomatic GPU kernel authoring in the Rust programming language. Extending the Rust ownership model to tile…

AI 摘要

NVIDIA 团队开发了一套 AI Agent 技能,用于把 cuTile Python 与 Triton-TileIR 编写的 GPU 内核翻译为 cuTile Rust。借助该流程,团队将 TileGym 全部 24 个公开算子迁移至 Rust,平均性能达到 Python 版本的 99.5%。该多智能体流水线覆盖分析、设备内核、主机与 FFI 代码及基准测试,每阶段均以可机器校验的判定收尾,并通过对比 Tile IR 验证翻译结果。

AI 问答

为什么要把 cuTile Python 内核翻译成 cuTile Rust?

cuTile Python、Triton-TileIR 与 cuTile Rust 都是同一 CUDA Tile IR 的前端,共用 tileiras 编译器。因此迁移不是重新做优化,而是在更安全的主机语言中重新表达相同的 tile 程序,编译器与性能模型保持不变,翻译结果也可校验。

翻译后的 Rust 内核性能表现如何?

团队使用该 AI Agent 技能把 TileGym 全部 24 个公开算子迁移到 cuTile Rust,包含约 40 个 GPU 内核,覆盖逐元素运算到 flash-attention decode、MLA 和 MoE,平均达到 cuTile Python 性能的 99.5%。

如何验证翻译的正确性?

通过转储参考内核与翻译内核的 Tile IR 并进行差异对比,可在运行任何测试之前结构性检查内存操作类型、tile 形状与归约是否一致,从而发现测试覆盖不到的错误。

打开原文

更多资讯

今晚「鸽派加息」?金色财经 · 09/16 16:08AI for everyone in every languageGoogle AI Blog · 09/16 00:00无问芯穹联合清华、上交正式开源具身端侧推理引擎APXInf,Pi 0.5性能SOTA量子位 · 09/15 18:43理想落幕:加密行业为何集体走向同质化?金色财经 · 09/15 18:31工作流可以自我进化了,英伟达开源 SoL-Pi,每小时省13.5刀!雷峰网 · 09/15 11:02F-Droid 上的应用有多少是在 AI 帮助下编写的?Solidot · 09/15 20:45Grayscale:比特币能否优化私募市场投资组合?金色财经 · 09/15 09:10Xspark AI 丁文伯:触觉替代不了视觉,但机器人需要一套自己的“脊髓” |物理AI 50人雷峰网 · 09/15 15:52Denuvo 起诉黑客违反 DMCA 反规避条款Solidot · 09/16 23:27长期利率为何不愿下跌?金色财经 · 09/15 18:04香港第一、全球第八:OSL进化背后的数字资产新格局金色财经 · 09/16 21:42豆包手机明日开卖,AI能帮你操作App了吗?36氪 · 09/15 22:32CoinEx宣布体面关停:值得肯定但不该神化金色财经 · 09/16 22:20一边喊停一边敲钟:硅谷AI巨头到底在怕什么金色财经 · 09/16 10:53基元律动与无问芯穹达成战略合作,推进高质量Token供给与应用雷峰网 · 09/16 16:20“裁判”不想给阿莫迪吹暂停金色财经 · 09/15 13:07Ctenophores Aren’t Just Beautiful. They’re Biological Wonders.Quanta Magazine · 09/16 23:56梁文锋CFO到位!投过智谱MiniMax量子位 · 09/15 21:53强化学习大本营新作:如何破解「学新忘旧」困局雷峰网 · 09/15 10:18英伟达开始限制Claude使用了金色财经 · 09/15 12:37想把喜欢的故事做成动画?我用 AIXTV 把《聊斋》做出来了人人都是产品经理 · 09/16 17:2236氪研究院 | 2026年北京市OPC创业者研究报告36氪 · 09/15 16:30前高通算法研发经理跨界AI算力板检测,获近亿元战略投资丨36氪首发36氪 · 09/15 09:30Build a serverless PII redaction pipeline with Amazon Bedrock Data AutomationAWS Machine Learning Blog · 09/16 23:17读完斯坦福116页AI实施手册,我最大的感受是:别再盯着模型了人人都是产品经理 · 09/15 12:14超级智能体“迪迪虾”上车,比亚迪携手阿里云推进智能座舱升级雷峰网 · 09/15 11:58协同办公进入Agent时代,飞书+豆包工作跑在了最前面量子位 · 09/16 15:46华为GTS让Agent学会「看着网络排障」,双防火墙难题几乎全拿下量子位 · 09/16 10:27一张GPU跑10万原子!分子之心用AI把化学反应“拍”成了电影量子位 · 09/15 17:58圆桌:中国之光成为新共识| 36氪 2026产业未来大会36氪 · 09/15 17:15