快捷导航
Quick Navigation
联系我们
英伟达 CUDA Tile IR 颁布发表开源:基于 MLIR 框架编
因为 CUDA Tile IR 采用 MLIR 做为根本,其理论上具备“降级”或转换到其他后端的可能性,从而为将来正在非 NVIDIA GPU 或加快器中支撑相关计较模子供给手艺根本。至多正在短期内,CUDA Tile IR 的开源将有帮于包罗 ZLUDA 正在内的相关兼容或移植项目标推进。
英伟达对 CUDA Tile IR 的描述是:“CUDA Tile IR 是一种基于 MLIR 的两头暗示和编译器根本设备,用于 CUDA 内核优化,,沉点支撑基于 tile 的计较模式,并针对 NVIDIA Tensor Core 单位进行优化。该项目为表达和优化面向 NVIDIA GPU 的分块计较供给了一个完整的生态系统,内存条理办理和 GPU 特定优化供给笼统,”。
MLIR 已被多家厂商和项目用于 AI 取高机能计较范畴:AMD 已正在其计较取 AI 软件栈中利用 MLIR,谷歌 IREE 项目同样依赖 MLIR 支撑多种硬件平台,英特尔也为自家硬件开辟了 MLIR“方言”XeVM。此外,ONNX-MLIR、MLIRE-AIE 等多种 IR 框架同样成立正在 MLIR 系统之上。
IT之家12 月 26 日动静,英伟达 12 月 4 日发布了全新的 CUDA 13。1 并引入了一种名为 CUDA Tile 的全新 GPU 编程体例,这是自 2006 年 CUDA 平台问世以来规模最大、内容最全的一次功能扩展更新。
按照英伟达申明,此次开源的 CUDA Tile 项目包含多个构成部门,包罗 Tile MLIR“方言”、当地 Python API 绑定、字节码暗示形式以及分歧性测试套件。

IT之家留意到,其相关代码已采用 Apache 2。0 许可证发布正在 GitHub 上,点窜和分发相关代码。