RuyiAI Triton-RISCV开源:在RISC-V上实现Triton原生编译

Triton-RISCV已在中国科学院软件研究所智能软件研究中心推出的RuyiAI系统软件栈中开源,欢迎大家体验和交流。相比现有的Triton项目,它有两个核心特点:

  • 原生支持RISC-V向量扩展(RVV)架构:可以在SG2044平台上直接原生构建与运行,无需交叉编译。

  • 复用buddy-mlir的向量化能力:在采用triton-shared将Triton转换成MLIR的Linalg方言后,buddy-mlir通过自定义的中间表示和下降通路进行优化。在SG2044单线程评测中,相比triton-cpu,最高性能提升达到4.16x,整体平均提升1.57x。

https://mp.weixin.qq.com/s/ux6lcT3qOb0uJrICOKRxnQ

1 个赞

好棒!我顺便给支持了一个新的后端 qemu-riscv64,可以在 x86 上也体验了 triton-riscv 了
zevorn/triton-riscv: Triton Compiler for RISC-V Platforms

1 个赞

给力 :clap: :clap: :clap:

1 个赞