机器之心·· 5 小时前AI 评分60
复旦与华为小艺提出 SAGE,昇腾 NPU 算子生成成功率达 95.5%
让AI学会给昇腾NPU调优!复旦联手华为小艺,算子生成成功率达95.5%
AI 导读
复旦大学与华为小艺团队提出 SAGE,把知识冷启动、工具增强诊断与经验学习串成一条流程,用于在昇腾 NPU 上生成并优化算子。在 NPUKernelBench 的 88 个算子上,SAGE 正确解出 84 个,成功率达 95.5%,相对 Value Memory 的 ER 提高 11.2 个百分点,自身优化加速中位数从 3.36 倍提高到 4.47 倍。
来源:机器之心 · mp.weixin.qq.com