跳到正文
arXiv Artificial Intelligence· Yupeng Chen, Kunxiang Zhao, Xingjian Wang, Yue Zhao, Yijing Wu, Ye Liu, Kai Li, Congliang Chen, Ruoyu Sun·· 18 小时前AI 评分38

TrainBridge:用 GPU 编译证据引导 NPU 训练优化的智能体方法

TrainBridge: Agentic NPU Training Optimization Guided by GPU Compilation Evidence

AI 导读

TrainBridge 将成熟 GPU 编译栈产出的优化执行作为任务参考,提取优化意图指导 Ascend NPU 训练程序改写,并通过迭代 profiling 精化。在覆盖 LLM 预训练、图像分类和视觉语言训练的 4 个负载上平均加速 1.7x,在 27 个 Modded-NanoGPT speedrun 记录上平均加速近 2.0x。

来源:arXiv Artificial Intelligence · arxiv.org