20210609-04
当前位置: 首页 > 滚动 >

OpenAITriton已开始合并AMDROCm代码,9月20日举行开发

来源: IT之家   浏览量:6244次   
2023-09-04 08:45:21

,Triton 是一种类似于 Python 的开源编程语言,它可以使没有 CUDA 经验的研究人员顺利编写高效的 GPU 代码,而且号称小白也可以写出与专业人士相媲美的代码,就是让用户用相对较少的努力实现最高的硬件性能,但 Triton 初期只支持英伟达 GPU。

OpenAI 声称:Triton 只要 25 行代码,就能在 FP16 矩阵乘法上达到与 cuBLAS 相当的性能。

从 Github 我们可以看到,OpenAI 已经开始在最新的Triton 版本中合并 AMD ROCm 相关分支代码,也正因此暴露了很多东西。也就是说,最新 Triton 后端已适配 AMD 平台,可谓意义重大。

通过官方的说法来看,他们已经通过了“test_core.py”上的大多数单元测试环节,但由于各种原因跳过了一些测试。

IT之家注意到,OpenAI 还宣布将于 9 月 20 日上午 10 点至下午 4 点在加利福尼亚州山景城的微软硅谷园区举行 Triton 开发者大会,而日程安排中就包括“将 Triton 引入 AMD GPU”和“Triton 的英特尔 XPU”两大环节,预计 Triton 将很快摆脱英伟达 CUDA 垄断的历史。

值得一提的是,Triton 是开源的,比起闭源的 CUDA,其他硬件加速器能直接集成到 Triton 中,大大减少了为新硬件建立 AI 编译器栈的时间。

在此前发布的 PyTorch 2.0 版本中,TorchInductor 便引入了 OpenAI Triton 支持,可为多个加速器和后端自动生成快速代码,同时实现用 Python 取代 CUDA 编程来写底层硬件的代码。也就是说,Triton 已经是PyTorch2.0 后端编译器关键构成部分。

实际上,此前 AMD ROCm 则主要是采用 Hipify 工具实现CUDA 兼容,而随着 AMD 开始为 RDNA 3 消费级显卡提供 ROCm 支持,预计后续将会有更多平台选择适配 AMD 硬件。

相关阅读:

《AMD 将为 RDNA 3 消费级显卡提供 ROCm 支持,从 Radeon RX 7900 XTX 和 Pro W7900 开始》

《AMD 正式推出 HIP SDK:拓展 ROCm 生态,支持运行英伟达 CUDA 应用》

广告声明:文内含有的对外跳转链接,用于传递更多信息,节省甄选时间,结果仅供参考,IT之家所有文章均包含本声明。

声明:以上内容为本网站转自其它媒体,相关信息仅为传递更多企业信息之目的,不代表本网观点,亦不代表本网站赞同其观点或证实其内容的真实性。投资有风险,需谨慎。

  • 精彩推荐
  • 东北虎豹等极度濒危物种在吉林实现种群恢复性增长东北虎豹等极度濒危物种在吉林实现种
  • 全球最大锂资源一体化利用项目开工 ​全球最大锂资源一体化利用项目开工
  • 华新环保:网上定价发行中签率为0.0285734496%华新环保:网上定价发行中签率为0.
  • 治水“下半场”:不鼓励各地追求过高的优良水体比例治水“下半场”:不鼓励各地追求过高
  • 清华大学施汉昌:水中微量污染物与生物传感技术的发展清华大学施汉昌:水中微量污染物与生
  • 福州:河湖长治久清 海绵全域铺开福州:河湖长治久清 海绵全域铺开
  • 要闻推荐
  • 图片新闻
  • 友情链接