用途:面向 Ascend C 算子开发与优化,提供从性能排查到代码修改、精度验证、性能对比的完整闭环流程,触发关键词包括 Ascend C 性能优化、算子优化、tiling、流水、搬运、内存优化、NPU/昇腾等。
主要能力:按六个阶段推进。Phase 1 排查:先读算子设计文档与 op_host、op_kernel 源码,再按 Tiling、搬运、API 使用、内存、流水五个阶段逐项检查,各阶段加载对应 reference 文件(如 references/tiling-prof.md、data-copy-prof.md、api-usage-prof.md、memory-prof.md、pipeline-prof.md),并输出按预期收益排序的排查报告。Phase 2 基线:确认或生成性能用例与对比报告,保存基线快照。Phase 3 优化:先加载 code-gen reference,再制定并执行修改方案,遵守反模式清单,随后编译安装。Phase 4 精度:调用精度评估流程验证功能正确。Phase 5 性能:用相同用例重新评估并与基线、标杆对比。Phase 6 迭代:未提升则继续,最多 3 轮。
使用方式:按阶段执行,修改前必须学习设计文档与 code-gen 知识;编译安装命令为 source ${ASCEND_HOME_PATH}/set_env.sh、cd task/ascend-kernel、bash build.sh、pip install output/ascend_kernel*.whl --force-reinstall --no-deps。产物包括性能用例、基线报告、性能报告、精度报告与优化汇总报告。
适用场景:昇腾 NPU 上 Ascend C kernel 算子的性能诊断与优化。限制:需具备对应算子工程与测试环境;迭代上限 3 轮;具体收益与平台细节以实际资料为准,暂未提供更多信息。