跳到正文
arXiv cs.AI· Yupeng Chen, Kunxiang Zhao, Xingjian Wang, Yue Zhao, Yijing Wu, Ye Liu, Kai Li, Congliang Chen, Ruoyu Sun·· 2 小时前AI 评分41

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

TrainBridge: Agentic NPU Training Optimization Guided by GPU Compilation Evidence

AI 导读

TrainBridge 提出一种智能体方法,将成熟 GPU 编译栈的优化经验迁移到 Ascend NPU 训练:从 GPU 训练执行中提取优化意图,指导 NPU 程序重写,并通过迭代 profiling 进一步优化。实验显示,四个涵盖 LLM 预训练、图像分类和多模态训练的工作负载平均加速 1.7x,在 27 项 Modded-NanoGPT speedrun 记录上平均加速接近 2.0x。

来源:arXiv cs.AI · arxiv.org