为 MNN 框架新增算子。包含 Schema 定义、形状计算、几何计算、后端实现、单元测试的完整 TDD 流程。分 5 步执行,每步有独立测试标准。
日本語の概要は準備中です。原文の説明を表示しています。
MNN CPU 后端(ARM / x86_64 / RISC-V 三侧)的总入口,只做分流不承载技术内容。下分 `optimize/`(为什么慢、该改哪一层)与 `kernel/`(这条 kernel 怎么写对、怎么被选中)两个分支,`shared/` 放两者共用的构建测试跑分命令、env 开关注册表与 RISC-V 开发板远端验证纪律。做 CPU 侧的工作但还不确定该进哪个分支,或需要三侧结构差异对照(第二张函数表按什么分、二级表怎么构造、`Precision_Low` 语义、ISA A/B 怎么做——「三侧不同构对照表」全树唯一一份,就在本文件)时读这里。
インストール方法を見るインストールする前に、エージェントに与えられる指示の中身を確認できます。
触发条件:CPU 上的性能优化或 kernel 开发,ARM / x86_64 / RISC-V 均适用。
本文件只做分流,不承载技术内容。先选分支,再进对应入口。 唯一的例外是下面「三侧不同构对照表」——那张表全树只有这一份,其他文档一律引用它、不许再抄。
| 你的问题 | 去哪 |
|---|---|
| 慢,还不知道原因,也不知道该改哪个文件 | optimize/SKILL.md → 从它的 diagnose-and-route.md 开始 |
| 要和别的推理框架比性能,或对手框架里找不到 MNN 融合算子的对应物 | optimize/diagnose-and-route.md §二(op 对 op、口径对齐、融合算子拆解) |
| 已有性能数据、已定位到某个 kernel,要动 SIMD / 汇编 / pack / 派发表 | kernel/SKILL.md |
| 「我到底跑在哪条 ISA 路径上」「为什么退到慢路径」 | optimize/arch/arm.md / optimize/arch/x86_64.md / optimize/arch/riscv.md |
| kernel 写完了,要决定跑哪些组合才算「过」 | kernel/correctness-gate.md |
| 结果不对,且出现在做完 CPU 性能改动之后,稳定地错在某一档 | optimize/bugfix.md |
| 结果不对,但同一输入每次跑都不一样 | general-debug/nondeterminism.md(§9 未初始化内存 / §10 多线程非确定) |
| 要跑构建 / 测试 / benchmark,或查某个开关的语义 | shared/build-test-and-benchmark.md / shared/env-registry.md |
| 这个算子 MNN 里还没有(缺 schema / shape / geometry) | add-new-op |
| RISC-V / RVV / SpacemiT IME2 厂商矩阵扩展 | 诊断 optimize/arch/riscv.md;实现 kernel/arch/riscv.md |
| 要在 RISC-V 开发板上编译 / 跑正确性 / 跑性能 | shared/riscv-remote-validation.md |
| 要跑 CI / 加测试阶段 / 真机 benchmark | test-ci |
最常见的走错:性能数据还没有就直接进 kernel/ 手写 kernel。手写 kernel 是最贵的一层,
先在 optimize/ 侧过投入决策门——已经跑到 82% roofline 的 kernel 再写一版汇编是白干。
skills/cpu/
├── SKILL.md ← 本文件,只分流
├── optimize/ 分支一:为什么慢,该改哪一层
│ ├── SKILL.md 分支入口
│ ├── diagnose-and-route.md bound 类型判定与路由(慢的第一站)
│ ├── runtime-and-scheduling.md L1/L2:线程数、ThreadPool、tiling、划分
│ ├── layout-and-memory.md L3:pack 格式、weight reorder、peak RSS
│ ├── bugfix.md 跨层交界处的正确性坑
│ └── arch/{arm,x86_64,riscv}.md L4 诊断面:我在哪条路径上 + 事故台账
├── kernel/ 分支二:这条 kernel 怎么写对,怎么被选中
│ ├── SKILL.md 分支入口(含四级实现阶梯与退出条件)
│ ├── pack-and-abi.md tile / cell stride / 后处理 ABI 契约
│ ├── dispatch-and-register.md 函数表注册、二级表安全写法、快照时序
│ ├── correctness-gate.md 跨 ISA × 精度 × 线程 × tail 的门禁
│ └── arch/{arm,x86_64,riscv}.md 实现面:目录命名、指令编码、寄存器分区
└── shared/ 两个分支共用的工具层
├── build-test-and-benchmark.md 构建开关、run_test.out、llm_demo/llm_bench、跑分纪律
├── riscv-remote-validation.md RISC-V 开发板:远端构建矩阵、板端正确性与性能实验
└── env-registry.md env / 编译宏 / backend flag / constexpr 四种机制
三个一级目录各自只有一种「东西」:optimize/ 是诊断,kernel/ 是实现,shared/ 是工具。
架构维度不构成一级目录,它是每个分支内部的 arch/ 子层——因为「ARM 的什么事」取决于你在诊断还是在实现,
这两件事的内容完全不同(诊断面只讲路径与自证,实现面只讲怎么写对)。
两处结构约定,不要去「改齐」:
optimize/arch/ 与 kernel/arch/ 各有三份、一一对应(arm / x86_64 / riscv)。
同一架构在两侧的内容完全不同:诊断面只讲路径与自证,实现面只讲怎么写对。
新增一个架构要么两侧都加,要么明确说明为什么只有一侧。shared/ 下只有工具,没有事实层。ISA 路径事实归 optimize/arch/(诊断视角),
注册与 ABI 事实归 kernel/。riscv-remote-validation.md 放在这里是因为它讲的是
怎么跑(远端构建矩阵、板端实验纪律),不是 RISC-V 的技术事实。
曾经放在顶层的 arm.md / x86_64.md 是分支拆分前的遗留,已按视角一分为二。| 分支 | 回答的问题 | 独占内容 |
|---|---|---|
optimize/ | 为什么慢,该改哪一层 | bound 类型判定与路由、L1/L2 线程与调度、L3 布局与 peak RSS、跨层不一致 bugfix、两侧派发路径诊断面与事故台账 |
kernel/ | 这条 kernel 怎么写对,怎么被选中 | 实现阶梯、pack/ABI 契约、派发表注册与二级表安全写法、跨 ISA × 精度正确性门禁、三份 ISA 实现参考 |
分界线是**「诊断 vs 实现」**,不是「性能 vs 正确性」:两个分支都要过正确性门禁, 但「运行时到底选了哪条路、为什么退到慢路径」属于诊断,「怎样正确注册使它能被选中」属于实现。
正确性内容按第二条轴分:事前门禁 vs 事后定位。 这条轴与「诊断 vs 实现」正交, 所以正确性文档在两个分支下各有一份,那不是重复:
| 你在做什么 | 去哪 | 它的产出 |
|---|---|---|
| kernel 写完了,要决定跑哪些组合才算「过」 | kernel/correctness-gate.md | 必测矩阵(六条会切换代码路径的轴)+ 两条判定标准 |
| 已经错了,要从「哪一档错」倒推到哪一层 | optimize/bugfix.md | 六类跨层不一致的机制、真实提交证据、预检清单 |
两份的症状表也按这条轴分工:交付前的静默失败症状在 correctness-gate.md §三,
事后的跨层症状路由在 bugfix.md §一。同一症状允许在两处各出现一行,
但必测取值只写在门禁里,机制与预检只写在 bugfix 里——不要在另一处复述。
同一个事实经常两侧都要提,规则是:诊断面只写「怎么看出来」,实现面只写「怎么写对」。
例如 Int8GemmKernelFast——「它为什么被选中」的判据写在 kernel/arch/,
「我这次到底选中了谁」的自证方法写在 optimize/arch/。
L1 Runtime 线程 / L2 Executor 调度 / L3 Layout 内存 / L4 Dispatch 函数表 / L5 Kernel ISA,
每层管什么、典型改动是什么、去哪份文档(含 L4/L5 的诊断面与实现面两个去处),见
optimize/diagnose-and-route.md §三——那张表全树唯一一份,本文不复述。
跨层交界处(L2↔L3 的 stride、L3↔L4 的 pack 与 tile、L4↔L5 的 ABI)是最贵的坑集中地,
症状多为「不崩、不报错、结果只是略差」,统一收在 optimize/bugfix.md。
不要把一侧的心智模型套到另一侧。 三侧不是同一个结构换了指令名,是五处根本不同:
| 维度 | ARM | x86_64 | RISC-V |
|---|---|---|---|
| 第二张函数表按什么分 | 精度:fp16 是另一张表 + 另一个 Backend(Arm82Functions / Arm82Backend),fp32 留在基表 | ISA:SSE 直接打进基表,AVX2/AVX512 才是另一张表 + 另一个 Backend(AVX2Functions / AVX2Backend) | 没有第二张表,也没有第二个 Backend:RVV 与 vendor 都在基表上逐字段覆盖 |
| 二级表怎么构造 | arm82 是逐字段赋值(约 120 条),漏字段 → nullptr 或不确定值 → 崩溃/乱码/随构建抖动 | new 后整体拷贝基表再打补丁,漏字段 → 继承 SSE 实现 → 慢但对 | 就地覆盖基表字段,漏字段 → 保留标量实现 → 慢但对 |
Precision_Low 语义 | 真 fp16:bytes=2、pack=8,是一条独立路径 | 不是 fp16,bytes 仍为 4;而且被 AVX2Backend 构造函数写死,用户请求的精度无效 → precision 轴在 x86_64 上是死的 | 没有 RVV fp16 路径,precision 轴不切换函数表 |
| ISA 档之间怎么做 A/B | MNN_CPU_TARGET 降档(需 -DMNN_PIPELINE_PROFILE=ON),clamp 0..3 | 同左,clamp 0..4 | 只能用两个独立 build 目录:MNN_RVV_SPACEMIT_IME2 ON/OFF 两个注册 TU 定义同名符号,是构建期互斥,没有运行时开关 |
MNN_CPU_USE_DEFAULT_BACKEND | 不影响 fp16(fp16 分支在它之前) | 静默绕过整条 AVX2/AVX512,pack 从 8/16 掉回 4 | 不改路径(它后面没有第二个 Backend),只是跳过多线程初始化 |
推论,三侧都常被踩:
MNN_CPU_TARGET 降档,而它默认构建下是彻底空操作(getenv 与能力位屏蔽整段都被
#ifdef MNN_PIPELINE_PROFILE 包住),必须 -DMNN_PIPELINE_PROFILE=ON 重建。MNN_RVV_SPACEMIT_IME2 后增量构建就比性能。坐标与自证方法分别在 optimize/arch/arm.md §二/§三、
optimize/arch/x86_64.md §二/§三 与
optimize/arch/riscv.md §二/§三;注册面的安全写法在
kernel/dispatch-and-register.md §三。
| 场合 | 写什么 |
|---|---|
| 平台 / ISA 术语(正文、结论、commit message) | x86_64、AArch64、RISC-V、RVV |
| 目录名、宏名、CMake 选项名(引用时保持字面) | source/backend/cpu/x86_x64/、cpu/arm/arm64/、cpu/riscv/common/、cpu/riscv/rvv/、cpu/riscv/spacemit_ime2/、MNN_USE_SSE、MNN_AVX2、MNN_AVX512、MNN_AVX512_VNNI、MNN_X86_USE_ASM、MNN_USE_ARMV82、MNN_SME2、MNN_USE_RVV、MNN_RVV_SPACEMIT_IME2、MNN_RVV_MARCH |
x86_x64、arm64、riscv 是仓库里的既有目录名,不要"修正"它们;反过来也不要在正文里写 x86_x64 或 riscv 当术语。
非平凡任务结束后,如果产生了可复用的教训,走 retrospective:
结论上提到对应的层文档或 optimize/bugfix.md,实验过程与原始数字不进本仓。
まだレビューはありません。使ってみた感想をお寄せください。
概要と使いどころ
为 MNN 框架新增算子。包含 Schema 定义、形状计算、几何计算、后端实现、单元测试的完整 TDD 流程。分 5 步执行,每步有独立测试标准。
日本語の概要は準備中です。原文の説明を表示しています。
MNN 各类正确性/回归 bug 的排查入口,按 bug 类别分册组织,本文件只做症状分流。分册:内存别名与生命周期(arena reuse、`MemChunk`、融合引入的别名竞争)、量化误差与导出侧权重损坏(低 bit 打包、导出分块、PyTorch MPS/CUDA 大张量静默错误)、host 侧并发/线程竞争(共享所有权的引用计数被写坏、析构链崩溃、TSAN A/B 与编译期哨兵)、fp16 表示能力不足(长序列复读、position 塌缩,以及「实时计算→预计算查表」重构的三类陷阱)、GPU shader 越界与 command buffer 故障、后端 kernel 隐式假设违反(causal mask、layout 约定)、持久化缓存误信(weight-mmap sync 自我污染、跨模型缓存复用)、逐 run 不同的非确定性(未初始化内存/堆垃圾依赖、多线程动态分发×异构 kernel)。用户报告 MNN 输出乱码/退化、单测或 golden 对不上、改动后回归、换后端结果不同、开某开关才错、结果每次跑都不一样、或崩在析构链上且只在后台线程异步释放时偶现时使用。
日本語の概要は準備中です。原文の説明を表示しています。
MNN CPU 后端 kernel 开发分支(`skills/cpu/` 下,另一分支是 `cpu/optimize` 性能归因)。覆盖标量 oracle → C++ SIMD → intrinsic → 汇编的四级实现阶梯、pack/kernel ABI 契约(tile、cell stride、后处理参数)、CoreFunctions 派发表注册与二级表安全构造、跨 ISA × 精度的正确性门禁,以及 AArch64 / x86_64 / RISC-V 三份实现参考。为 CPU 后端新写或移植 kernel(NEON / SSE / AVX / RVV intrinsic 或 .S 汇编)、新增一层 ISA、改 pack mode 或 tile 参数、把 kernel 挂进函数表时使用。已定位到 kernel 才进本分支。
日本語の概要は準備中です。原文の説明を表示しています。
MNN CPU 后端性能归因分支(`skills/cpu/` 下,另一分支是 `cpu/kernel` kernel 开发)。按五层(Runtime 线程 / Executor 调度 / Layout 内存 / Dispatch 函数表 / Kernel ISA)定位瓶颈,含 bound 类型判定、op 级实验回路、跨框架 op 对 op 对比、跨层不一致的事后定位,以及 ARM / x86_64 / RISC-V 三侧「我到底跑在哪条 ISA 路径上」的诊断面。CPU 上算子或 LLM prefill/decode 慢、线程数或内存占用异常、出现性能回归、要与外部推理框架逐算子对比时使用。
日本語の概要は準備中です。原文の説明を表示しています。
MNN Hexagon/HVX/HMX DSP 后端(`source/backend/hexagon`)的优化、重构、构建与回归验证。覆盖设备实测的相位分解、测量纪律、v79/v81 双架构差异、常见瓶颈模式、已否证方向与 cDSP crash 诊断。
日本語の概要は準備中です。原文の説明を表示しています。
MNN Metal 后端 op/kernel 开发与优化入口。索引各份 sub-doc:性能问题诊断流程(op 单测 + 对手基准定位真瓶颈,优化任务第一站)、kernel 开发规范与优化知识库(命名/写法/GEMV/GEMM/attention + 手段方法论)、算子融合全链路(导出图→converter→Metal 单 dispatch + 融合方法论)、运行时调度(fence/content-cache/H2D/replay + 调度方法论)、构建测试基线、env 开关注册表。根据当前任务选择性阅读对应 sub-doc。
日本語の概要は準備中です。原文の説明を表示しています。