microsoft/antares

这是什么？

核心功能

• 01允许用户以 CUDA 或 HIP C++ 语法编写 PyTorch 算子内核。
• 02提供 autort 命令行工具，可编译并运行命令中定义的内核。
• 03支持在简短的声明中指定张量形状和线程绑定。
• 04包含面向 AI 智能体编写内核的 SKILLS.md 指南。
• 05提供 CUDA 和 ROCm 环境的 Docker 配置。

适用场景

• 在 NVIDIA GPU 上原型验证自定义 PyTorch 算子。
• 借助 ROCm 为 AMD GPU 移植或调优算子。
• 让 AI 智能体生成并测试加速器内核。

快速上手

1. 1

   使用推荐的 Docker 镜像搭建环境（CUDA 或 ROCm）

2. 2

   使用 autort 命令运行 Hello World 示例

   autort -n hello_world_example --source=' @DEF_FUNC: input:float32[B, M] -> output:float32[B, M] @DEF_BIND: ~B~:1, ~%~:1024 void main() { if (int(blockIdx.x) == 0 && int(threadIdx.x) == 0) printf("(autort) Hello World!\n"); }'

使用前须知

• !README 未提供包管理器安装命令，安装依赖于推荐的 Docker 镜像。
• !需要 CUDA 或 ROCm 加速器环境。
• !README 指向的手册和旧版文档已弃用。

收录于专题

标签

常见问题

antares 是什么？

AutoRT 是一款编译器工具，帮助开发者为自有加速器创建、评测并优化 PyTorch 算子。 AutoRT 是微软 Antares 项目的新一代版本，是一款面向运行时用户的编译器方案，用于为 PyTorch 在自有加速器上编写、评测和调优算子。内核以 CUDA 或 HIP C++ 函数体的形式编写，并通过简短的声明指定输入输出张量形状和线程绑定。

antares 可以免费使用吗？用的什么许可证？

antares 以 Other 许可证开源。用于商业产品前，请先确认许可证条款。

antares 还在维护吗？

活跃维护中。antares 最近一次提交是在 2026年10月23日。

antares 适合用来做什么？

在 NVIDIA GPU 上原型验证自定义 PyTorch 算子。 借助 ROCm 为 AMD GPU 移植或调优算子。 让 AI 智能体生成并测试加速器内核。

antares 怎么安装和运行？

使用推荐的 Docker 镜像搭建环境（CUDA 或 ROCm） 使用 autort 命令运行 Hello World 示例: autort -n hello_world_example --source=' @DEF_FUNC: input:float32[B, M] -> output:float32[B, M] @DEF_BIND: ~B~:1, ~%~:1024 void main() { if (int(blockIdx.x) == 0 && int(threadIdx.x) == 0) printf("(autort) Hello World!\n"); }'

在哪里获取 antares？

源代码和安装说明在 GitHub 上：https://github.com/microsoft/antares