OpenAI Triton 已开始合并 AMD ROCm 代码，9 月 20 日举行开发者大会

图片地址：https://img.ithome.com/newsuploadfiles/2023/9/d810e53c-7fcd-4115-88d8-8c8f0f6afd11.png?x-bce-process=image/watermark,image_aW1nL3dhdGVybWFyay9xYy9xYzEwOS5wbmc=,t_100,g_9,y_19,x_19,a_0

IT之家 9 月 3 日消息，Triton 是一种类似于 Python 的开源编程语言，它可以使没有 CUDA 经验的研究人员顺利编写高效的 GPU 代码（可以理解为简化版 CUDA），而且号称小白也可以写出与专业人士相媲美的代码，就是让用户用相对较少的努力实现最高的硬件性能，但 Triton 初期只支持英伟达 GPU。OpenAI 声称：Triton 只要 25 行代码，就能在 FP16 矩阵乘法上达到与 cuBLAS 相当的性能。从 Github 我们可以看到，OpenAI 已经开始在最新的 Triton 版本中合并 AMD ROCm 相关分支代码，也正因此暴露了很多东西。也就是说，最新 Triton 后端已适配 AMD 平台，可谓意义重大。通过官方的说法来看，他们已经通过了“test_core.py”上的大多数单元测试环节，但由于各种原因跳过了一些测试。IT之家注意到，OpenAI 还宣布将于 9 月 20 日上