Alibaba:GitHub 新仓库· alibaba·· 2026-06-18AI 评分38
阿里巴巴发布 atrex-bench:AI 生成 GPU kernel 的端到端基准
alibaba/atrex-bench
AI 导读
阿里巴巴开源 atrex-bench,一个面向 AI 生成 GPU kernel 的端到端基准,取材自真实生产 trace。它要求把 PyTorch 参考实现改写为 Triton、Gluon、FlyDSL、CuteDSL 等 DSL kernel,并从编译、数值正确性和 speed-of-light 效率三个维度评分,同时覆盖 AMD 与 NVIDIA GPU。
来源:Alibaba:GitHub 新仓库 ·