Triton 是 OpenAI 推出的类似于 Python 的开源编程语言,用于编写高效自定义深度学习原语的语言和编译器。
Triton 的目标是提供一个开源环境,以比 CUDA 更高的生产力编写快速代码,同时比其他现有 DSL 具有更高的灵活性。
该项目的基础在以下 MAPL2019 出版物中进行了描述:Triton: An Intermediate Language and Compiler for Tiled Neural Network Computations。
支持的平台:
- Linux
支持的硬件:
- NVIDIA GPU(Compute Capability 7.0+)
- 开发中:AMD GPU、CPU
评论