跳到正文
AI HOT 精选

摩尔线程开源 MusaCoder,用国产 GPU 集群训练出能自动写底层算子代码的模型

摩尔线程开源 MusaCoder 代码大模型,9B/27B 参数基于国产 GPU 全链路训练

摩尔线程放出了 MusaCoder,一个专门用来生成 GPU 底层算子代码的模型,有 9B 和 27B 两个版本。它的完整训练流程全跑在自家的 MTT S5000 集群上,没依赖国外硬件。这个模型的作用是,你给它一个 PyTorch 的标准算子,它能自动写出对应的 CUDA 或 MUSA 原生高性能代码,省去手写底层代码的麻烦。在 KernelBenc...

推荐理由:摩尔线程开源了 MusaCoder,一个用自家 MTT S5000 集群从头训练的代码模型,专门把 PyTorch 标准算子自动转成 CUDA 或 MUSA 原生高性能代码。这件事的看点在于全链路国产化——芯片、训练、模型输出都是自己的,不是拿开源模型微调。我会先打个折:正文没披露训练成本、数据规模和具体延迟,KernelBench 的分数也没给全,所以性能到底多强还不好说。但如果是真的,对做国产 GPU 算子开发的人挺省钱。

读原文 ↗导出 Markdown