CANN工具脚本使用指南

Tools Summary

【免费下载链接】cannbot-skillsCANNBot 是面向 CANN 开发的用于提升开发效率的系列智能体，本仓库为其提供可复用的 Skills 模块。项目地址: https://gitcode.com/cann/cannbot-skills

Scope

• agent/scripts/estimate_matmul_datamove.pyis a lightweight estimator for matmul data movement and tile-space feasibility.
• agent/scripts/build_agent_index.pybuilds machine-readable agent indexes from the human-readable example catalogs underagent/references/examples/.
• agent/scripts/select_kernel_example.pyranks existing kernel examples from the generated kernel index so authoring can start from the right study target faster.
• agent/scripts/check_kernel_catalog.pychecks thatkernel-catalog.md, the generated kernel index, and the realagent/example/kernels/*.pyfiles still agree.
• agent/scripts/gen_kernel_skeleton.pygenerates repository-style kernel scaffolds for cube-only, cube->vec, vec->cube, vec->cube->vec, cube->vec->cube->vec, and vec->cube->vec->cube topologies, with topology-specific profile variants.
• These tools do not launch kernels. They evaluate metadata, formulas, search candidates, or strategy estimates in Python.

Core APIs

• estimate_percore_datamove(m, n, k, TILEM, TILEN, TILEK, mode, dbuf_left=True, dbuf_right=True, dbuf_l0c=True)
  • Estimates single-core data movement for one fixed loop mode.
  • Validmodevalues areleft_first,right_first, andbalanced.
  • Both operand-tile capacity andL0Coutput-tile capacity are checked before returning a result.
• estimate_multi_core(m, n, k, m_split, n_split, TILEM, TILEN, TILEK, nonempty_only=False, dbuf_left=True, dbuf_right=True, dbuf_l0c=True)
  • Converts the full problem into a per-core subproblem, tries all three loop modes, and returns the minimum total data movement.
  • nonempty_only=Truecounts only non-empty split blocks;Falsemultiplies by the full split grid.
• estimate_strategy(m, n, k, num_core, split_mode, min_tile_m=None, min_tile_n=None, dbuf_l0c=True)
  • Brute-force searches split, tile, mode, and legal DBUF combinations.
  • Returns a dictionary withbaseline_datamove,best_datamove,best_results, and a PrettyTable object intable.

Capacity Model

• Tile-space feasibility usesMAX_TOTAL_TILE_ELEMENTS = 128 * 1024.
• dbuf_left=Falsehalves only the left-matrix space term.
• dbuf_right=Falsehalves only the right-matrix space term.
• L0Coutput-tile feasibility is checked separately with:
  • dbuf_l0c=True:TILEM * TILEN <= 32 * 1024
  • dbuf_l0c=False:TILEM * TILEN <= 64 * 1024
• DBUF flags affect capacity only; they do not change the contenteditable="false">【免费下载链接】cannbot-skillsCANNBot 是面向 CANN 开发的用于提升开发效率的系列智能体，本仓库为其提供可复用的 Skills 模块。项目地址: https://gitcode.com/cann/cannbot-skills