开源项目

NVIDIA TensorIR:将张量计算编译为CUDA Tile计算核

NVIDIA/tensor-ir

AI 基础设施模型部署

概述

TensorIR提供张量计算的编译前端。它将MLIR表示的计算转换到CUDA Tile路径,供开发者构建和实验。仓库项目用途依据公开文档介绍;具体支持范围及安装步骤按对应文档确认。项目处于早期阶段,应确认支持的算子、工具链和硬件;不能将预览实现当作生产性能承诺。