结论：codegen/simplify.py 处理 UOp 图中的循环范围、条件索引和归约，是编译器内核优化插件，不是可在线替换的控制插件。它的正确性靠几道静态门保证，最核心的是：合并范围只接受 FLOORDIV/FLOORMOD 总数不增加的结果；RANGE % 常量 拆分必须能静态证明整除、且范围长度节点本身为 CONST；遇到无门索引、STORE 或嵌套 REDUCE 时收缩或 collapse 会被阻止。源文档另指出 postopt symbolic 不受 if optimize 保护，插件输入必须是已通过类型、范围及缓冲 ABI 校验的 UOp 图。
编译调用链：optimize 与非 optimize 的分界
get_kernel_graph() 先 run_rangeify()，再用 symbolic + pm_reduce_simplify 等规则简化无负载归约，随后把缓冲图降到 STORE。full_rewrite_to_sink() 仅在 optimize=True 时依次执行 pm_load_collapse、pm_split_ranges + pm_flatten_range、初始符号化、pm_flatten_range + pm_simplify_ranges，再进入后范围优化；关闭优化并非仍会执行这些规则。
源文档特别指出：同一函数的 postopt symbolic 不受上述 if optimize 保护，仍运行 pm_reduce_unparented；而 pm_reduce_load_collapse 是供 pm_load_collapse 内部使用的增强规则集，不能当成独立顶层 pass。
范围规则与关键函数
| 函数或规则 | 作用及必要条件 |
|---|---|
flatten_range() |
对 REDUCE/END 的普通范围经 SINK 重新收集并展平；void/bool 回边留在末尾，不作为普通循环范围结束。 |
simplify_merge_adjacent() |
END 只试相邻范围，REDUCE 试排列组合；两轴必须同 AxisType、在所有内部 REDUCE 中归属一致；合并后的索引用商/余数还原，且只接受 FLOORDIV/FLOORMOD 总数不增加的结果；每次只合并一对。 |
mark_gated() |
从 INDEX 的 WHERE 有效条件中识别 RANGE < CONST，对同一范围取已见上界的最大值；若某次索引未受该范围的门保护，就记录原范围长度；REDUCE 也强制原长度。 |
pm_simplify_ranges |
汇集 END/REDUCE 合并、INDEX 门收集、REDUCE 禁止收缩；直到 SINK 才通过 do_substitute() 一次性替换范围并清空上下文。 |
mark_range_mod() |
只针对 RANGE % 常量；范围长度须为常量且 UOp.divides(c.val) 能证明被模数整除；WARP/DEVICE 轴排除；已有待拆轴不会重复登记。 |
pm_split_ranges |
在 SINK 用商轴与余数轴重建原索引：outer * c + inner，新增轴编号由原 arg 派生。 |
reduce_unparented() |
源表达式未引用的 REDUCE 轴可移除：ADD 乘以轴长，MUL 对轴长取幂，MAX 保持原值；其他归约运算不走此规则。 |
源文档强调：这里的「整除」是静态证明，不是运行时猜测。UOp.divides() 对常量、STACK、ADD、MUL 和带 multiple_of 约束的 PARAM/BUFFER 做有限证明，无法证明即返回 None；mark_range_mod() 还要求范围长度节点本身是 CONST，所以动态长度即使数学上可能整除，也不会触发该拆分。
四个正确性门
| 正确性门 | 表现 |
|---|---|
| 不增加 FLOORDIV/FLOORMOD 数量 | 合并相邻范围后，只接受商/余数总数不增加的结果。 |
| 归约范围归属一致 | 两个轴必须在所有内部 REDUCE 中归属一致，否则不能合并。 |
| 无门索引和归约阻止收缩 | 某次索引未受范围门保护时记录原长度，REDUCE 强制原长度。 |
| STORE 和嵌套 REDUCE 阻止 collapse | reduce_collapse() 发现 STORE 或内层 REDUCE 即退出。 |
pm_reduce_collapse 包含无父范围规则和几类代数重写：把与 RANGE 无关的加项或正整数乘项移出小于比较；把有上下界的布尔门求和化为截断区间长度；拆分加法求和；把布尔乘法改为 WHERE。源文档指出适用范围由模式和 no_range() 判断决定，不意味着所有求和都能闭式化简。
pm_reduce_load_collapse 额外识别「用动态索引挑选某个 RANGE 元素」的 gated load 求和：替代 RANGE 前先检查索引在 [0, extent)，并保留有效性条件；其最后一条规则在含加载索引的比较中逆转一种代数移项，以免对加载得到的索引做可能溢出的加法。源文档提醒，不能把这一优化等同于任意 gather 的安全实现。
源文档还指出 reduce_collapse() 的处理方式：它逐归约轴提取依赖子图，外部输入先换成有范围界的临时 PARAM 再运行模式重写，结果仍含 RANGE 时放弃。这一设计避免把外部动态表达式误当成轴内常量，但只证明匹配规则覆盖的形式，不等于对所有求和都成立。
面向硬件与插件化
这层适合定义为编译器内核优化插件。插件输入应为已通过类型、范围及缓冲 ABI 校验的 UOp 图；输出须保留 RANGE 的轴类型与索引等价、门条件、归约 dtype/溢出语义以及 LOAD/STORE 副作用。ctx 是一次图重写的临时状态，不能跨模型或并发编译复用。
车载推理部署应将编译器版本、模型工件、renderer/设备能力、缓冲布局及测试结果锁定在同一版本。关键回归包括非整除/动态轴长度、无门索引、归约嵌套、边界像素、整数溢出、不同后端浮点归约误差和最坏执行时延。源文档明确：编译通过不证明相机输入、模型输出或整车控制安全，执行许可仍由车型适配和 panda 固件安全门约束。
验证与限制
上游四项定向用例已通过：TestReduceCollapse.test_multi_range_reduce_add、test_reduce_shapeless_const_unroll、TestSchedule.test_arange_sum、test_arange_sum_alt（PYTHONPATH=. DEV=PYTHON）。test_simplify_valid_idx.py 的范围收缩用例因环境缺少 z3 而未运行。源文档明确未进行完整后端编译、LFS 模型数值、目标 GPU、实车或 CAN 验证，source/ 未修改。
常见问题
这层化简是可在线替换的控制插件吗？
不是。源文档把它定义为编译器内核优化插件，不是可在线替换的控制插件：插件输入应为已通过类型、范围及缓冲 ABI 校验的 UOp 图，执行许可仍由车型适配和 panda 固件安全门约束。
RANGE % 常量为什么不是总能拆分？
因为它要求静态证明。mark_range_mod() 要求范围长度是常量，且 UOp.divides(c.val) 能证明被模数整除；UOp.divides() 只对常量、STACK、ADD、MUL 和带 multiple_of 约束的 PARAM/BUFFER 做有限证明，无法证明即返回 None，动态长度也不会触发拆分。
为什么不把加载索引的求和都做闭式化简？
因为需要保留边界条件。源文档指出 pm_reduce_load_collapse 只在替代 RANGE 前确认索引在 [0, extent) 并保留有效性条件，且最后一条规则刻意避免对加载索引做可能溢出的加法；不能把这一优化等同于任意 gather 的安全实现。
编译器优化通过能否说明推理安全？
不能。源文档明确编译通过不证明相机输入、模型输出或整车控制安全；车载部署应把编译器版本、模型工件、后端能力与测试结果锁定在同一版本，并覆盖溢出、动态轴长度与最坏执行时延等回归。
延伸阅读
- tinygrad 张量调度与内存 —— 化简所在的调度与缓冲图上游。
- tinygrad dtype 与 UOp 降级 —— 同一 UOp 图上的另一类重写与语义保持。
- tinygrad JIT 捕获与重放 —— 编译产物如何被捕获与复用。
- tinygrad 网络层状态与形状契约 —— 形状与轴语义在上层的表达。
参考资料
- 项目 02 ——
openpilot-research/TINYGRAD_RANGE_REDUCE_SIMPLIFY_PLUGIN_CONTRACT_DEEP_DIVE.md（取用了编译调用链分界、范围规则表、四个正确性门、插件化输入输出约束与定向用例结果）。
说明：本文为对固定源码快照的静态结构分析，未运行完整测试套件、未连接硬件、未验证目标车型的真实参数组合。文中所有阈值、行号索引与配置均出自源文档，标注为未验证的结论保留其原有边界。
更新记录
首次发布 2026-10-11 最近核对：2026-10-11
发表回复