OpenAI 开源新的 AI 编程语言,以替代 Nvidia CUDA
OpenAI 宣布推出一种类似于 Python 的开源编程语言Triton,并发布了Triton 1.0 版本。根据介绍,Triton 可使没有 CUDA 经验的研究人员能够编写高效的 GPU 代码,且大多数情况下可以与专家所能编写的代码质量媲美。 OpenAI 称,Triton 使得以相对较少的努力达到峰值硬件性能成为可能。例如,它可以用不到 25 行代码编写与 cuBLAS 性能相匹配的 FP16 矩阵乘法内核,这是许多 GPU 程序员无法做到的。“我们的研究人员已经使用它来生成效率比同等 Torch 实现高 2 倍的内核,我们很高兴与社区合作,使 GPU 编程对每个人来说都变得更加容易。” OpenAI 指出,一些在使用 CUDA 框架时被认为相当有挑战性的问题都在 Triton 中得到了解决。与 CUDA 中的编码相比,Triton 可以为一些处于 AI 机器学习形式核心的神经网络任务(例如矩阵乘法)提供显着的易用性优势。Triton 语言会自动执行许多 AI 代码优化,以节省开发人员的时间;其拥有的相对简单性还可以使没有丰富 CUDA 编程经验的软件团队能够创建比其他方法更高...
