跳到正文
原文
AITNT · AI头条(网页)·· 3 小时前AI 评分64

复旦大学联手华为小艺提出 SAGE 框架,实现昇腾 NPU 算子自动生成与性能调优

让AI学会给昇腾NPU调优!复旦联手华为小艺,算子生成成功率达95.5%

AI 导读

复旦大学与华为小艺团队联合推出 NPU 算子开发框架 SAGE,通过知识冷启动、工具增强诊断与经验自进化机制打通昇腾硬件代码的编译、纠错与性能优化流程。方案设计了采用追踪效用估计与效用门控巩固策略,将有效经验提炼为常驻规则并支持跨算子复用。在 Ascend 910C 上的 NPUKernelBench 等测试中,系统显著提高了算子生成正确率,并在多个复杂算子上跑出数倍至数十倍于基准实现的优化加速。

来源:AITNT · AI头条(网页) · aitntnews.com