用途:为 AscendC 自定义算子生成 README 风格技术文档。做法是让模型读取一份参考 README 和目标算子源码,再综合出格式匹配的文档;不使用 Python 解析脚本推断算子语义,只用 shell 搜索与读取工具定位文件、查看源码。
主要能力:从 host 注册文件提取算子名、输入输出、dtype 与 format、平台限制(如 ascend910b);从 host 与 tiling 头文件提取 tiling 字段、shape 推导、固定常量、workspace 公式、block 与 core 切分规则;从 kernel 文件提取运行时参数解析、主计算流水、mask 行为、reduction 行为、输出写回布局。
使用方式:先读取一份代表目标格式的参考 README,再按 basename 定位 op_host/<op_name>.cpp、op_host/<op_name>_tiling.h、op_kernel/<op_name>.cpp;若仓库无按算子分目录,则按算子名建文档路径,如 .../docs/<op_name>/README.md;文件归属不明时先查 CMakeLists.txt 或 rg 引用。文档结构建议为 Overview、Inputs And Outputs、Runtime Parameters、Tiling Design、Kernel Design、File Layout Mapping、Build Entry、Limitations、Summary。完成后重开 README,逐节对照源码核验。
适用场景:AscendC 算子仓库需要补充或统一技术文档时;仓库布局与模板不同、需按算子名归一化文档单元时;用户希望改用外部 API 时,同样先本地收集模板与源码,再发送紧凑提示与必要片段,把 API 当写作引擎而非解析器替代。
限制:代码未直接写明的内容须标注为推断,不得当作事实;代码未提供可运行命令或明确框架调用点时,应说明缺失而非编造;输出 shape 或辅助输出若为隐式,需说明其遵循实现契约并指出不确定性;不得声称仓库未展示的运行时行为、调用示例或测试覆盖。参考资料包括 references/readme-outline.md、references/source-review-checklist.md、references/model-prompt-template.md。