CANN-Bench

新AI方法使用知识库进行NPU内核生成

研究人员开发了ASCENDEVOLVE，一种新颖的NPU内核生成方法，该方法超越了仅依赖经验。这种新方法整合了一个大型外部知识库，该知识库源自平台文档和算子存储库，以指导内核搜索过程。通过索引和检索相关信息，ASCENDEVOLVE能够播种多样化的实现策略，辅助纠错，并优化候选内核。该系统在Ascend C算子的CANN-Bench基准测试中表现出显著的改进，性能优于Reflexion和OpenEvolve等现有方法。