描述

一、主要解决的问题

1.1 运行时换轴破坏 canonical 槽位语义

原实现通过交换 input.local_buffer_vars 中的指针选择 Reduce Tile 求解顺序。该数组同时被公式、约束、equal_order、PGO 候选和 tiling_data 按固定下标解释；交换 z1t/z2t 后会让轴身份与槽位含义不一致，存在回填错位和精度风险。

1.2 动态选择未完整作用于求解路径

原实现只交换两个变量，不能表达完整轴排列，也没有完整覆盖 WorkloadBalance 和 PGO/topn 的枚举顺序，动态决策无法稳定影响后续全部求解路径。

1.3 equal_order 与变量替换需要隔离

equal_order 依赖 AttAxis::order 的 canonical 分组；缺失、重复或多重 Expr 映射可能造成误分组。运行时 preferred order 中还可能包含常量或非 local-buffer Expr，需要先投影并校验。

二、修改方案

2.1 canonical 和 ordered 双视图

• local_buffer_vars 保持 canonical 顺序，固定用于轴身份、equal_order、公式、约束、性能计算和 tiling_data 回填。
• 新增 ordered_local_buffer_vars，仅用于普通 solver、WorkloadBalance 和 PGO/topn 的遍历顺序。
• 新增 ordered_to_canonical，将 ordered 下标映射回 canonical 下标；两种视图共享同一 TilingVariable，不复制状态。

2.2 生成完整运行时规则

• RuntimeReorderRule 使用原始 tail/reduce Expr、参数化阈值和完整 preferred_order。
• canonical 与 preferred 分别构建拓扑；只有 canonical 写入 AttAxis::order，preferred 只生成完整排列。
• equal-order 组内相对顺序保持 canonical；非法、缺失或重复排列不生成运行时重排代码，安全回退 canonical。

2.3 求解和输出隔离

• 运行时条件 tail < cache_line_size / dtype_size 且 reduce > vector_len_size / dtype_size 命中时，改写 ordered 视图及映射，不修改 canonical 数组。
• equal_order 始终在 canonical 轴上识别和联合求解；普通求解按 ordered 遍历，并通过映射访问 canonical solved_axes。
• PGO 按 ordered 枚举和重置变量，候选值按 canonical 下标写回；topn 过滤及最终 tiling_data 均保持 canonical 布局。

三、代码修改流程图

flowchart TD
    A[ArgListReorder 分析 Reduce Tile] ==> B{shape 是否动态且满足生成条件}
    B ==>|否| C[仅保留 canonical 拓扑]
    B ==>|是| D[独立生成 canonical 与 preferred 拓扑]
    D ==> E[canonical 写入 AttAxis order]
    D ==> F[preferred 提取完整 Tile Expr 顺序]
    F ==> G[保持 equal-order 组内 canonical 相对顺序]
    G ==> H{投影后是否为 local-buffer 完整全排列}
    H ==>|否| C
    H ==>|是| I[保存 RuntimeReorderRule]
    C ==> J[Codegen 初始化 canonical ordered 和单位映射]
    I ==> J
    J ==> K{运行时 tail 小且 reduce 大}
    K ==>|否| L[ordered 保持 canonical]
    K ==>|是| M[ordered 切换 preferred 并更新映射]
    L ==> N{求解路径}
    M ==> N
    N ==>|equal-order| O[canonical 分组和联合求解]
    N ==>|普通与 WorkloadBalance| P[ordered 遍历并映射 canonical index]
    N ==>|PGO 与 topn| Q[ordered 枚举并按 canonical 下标保存候选]
    O ==> P
    P ==> R[公式 约束 性能计算读取 canonical]
    Q ==> R
    R ==> S[tiling_data 按 canonical 槽位回填]

四、核心类图

classDiagram
    class RuntimeReorderRule {
        Expr condition_axis
        Expr compare_axis
        uint32 condition_threshold
        uint32 compare_threshold
        ExprVector preferred_order
    }
    class ModelInfo {
        RuntimeReorderRuleVector runtime_reorder_rules
        AttAxisVector arg_list
    }
    class ArgListReorder {
        SortArgList()
        SetRuntimePreferredOrder()
        GetTileSplitOrder()
    }
    class ArgsManager {
        GetAxesOrder()
        GetVarsRelations()
    }
    class SolverPassManager {
        InitSolverGen()
    }
    class AxesReorderSolverGen {
        SetAxesOrder()
        SetVarsRelations()
        GenRuntimeReorderRules()
        GetRuntimePreferredIndices()
    }
    class AxesReorderSolverInput {
        TilingVariablePointer local_buffer_vars
        TilingVariablePointer ordered_local_buffer_vars
        uint32Pointer ordered_to_canonical
    }
    class AxesReorderSolver {
        IdentifyEqualPriorityAxes()
        NaiveLocalBufTiling()
        BinaryLocalBufTilingCore()
        WorkloadBalance()
    }
    class AxesReorderPgoSolver {
        PgoSolverGenerateAllTilingDataInner()
    }
    class TilingVariable {
        size_t order
        int64 value
    }
    ArgListReorder ..> RuntimeReorderRule : 生成完整规则
    ModelInfo o-- RuntimeReorderRule : 保存
    ArgsManager ..> ModelInfo : 读取 canonical 信息
    SolverPassManager ..> ArgsManager : 获取 order 与变量关系
    SolverPassManager ..> AxesReorderSolverGen : 初始化 codegen
    AxesReorderSolverGen ..> RuntimeReorderRule : 解析 preferred order
    AxesReorderSolverGen ..> AxesReorderSolverInput : 生成双视图输入
    AxesReorderSolverInput o-- TilingVariable : 两种视图共享对象
    AxesReorderSolver ..> AxesReorderSolverInput : 普通求解
    AxesReorderPgoSolver ..> AxesReorderSolverInput : PGO 枚举

变更类型

• Bug 修复
• 新功能
• 代码风格更新
• 重构
• 构建过程或辅助工具的变动
• 文档内容更新

关联的Issue

无。

如何测试

一、测试用例说明

1.1 单元测试

• 完整 ATT UT：1022/1022 通过。
• 新增规则生成、完整排列、非法排列回退、替换变量一对一映射、equal_order 隔离、WorkloadBalance ordered 遍历、PGO canonical 回填等覆盖。
• 独立 review 定向用例：13/13 通过。

1.2 系统测试

• att_st：通过。
• inductor_topn_test_codegen：通过。
• inductor_tail_brc_tail_reduce_test_codegen：通过。
• pgo_add_abs_inductor_test_codegen：通过。

核对清单

• 我的代码遵循了项目的代码风格
• 我已对代码进行了自测
• 我已更新了相关的文档
• 我在标题中使用了合适的类型标签
• 我已经详细阅读了贡献指南（CONTRIBUTING.md）

其他信息

验证方法

• git diff --check 通过。
• changed-lines clang-format 检查通过，无额外格式化。
• OAT 对变更源码和测试文件检查通过。
• 构建产物时间晚于相关源码。
• 独立代码 review 通过，无 Critical 或 Important。

注意事项

• 本次不修改对外 API 或 ABI，不新增堆分配及设备侧逻辑。
• 动态 Reduce 的真实设备侧 E2E 覆盖未包含在本 PR，作为非阻塞后续增强项。

提交记录

Commit	描述	修改文件数
4cb7547d	修复 Reduce Tile 运行时轴重排	16

修改文件清单

文件路径	修改类型	说明
autofuse/att/base/model_info.h	修改	RuntimeReorderRule 改为完整 preferred order
autofuse/att/gen_model_info/expr_gen/arg_list_reorder.cpp	修改	生成并校验 canonical 与 preferred 拓扑
autofuse/att/gen_model_info/expr_gen/arg_list_reorder.h	修改	增加 preferred 顺序辅助接口
autofuse/att/generator/preprocess/args_manager.cpp	修改	生成 canonical axis order 及唯一 fallback
autofuse/att/generator/preprocess/args_manager.h	修改	暴露 GetAxesOrder
autofuse/att/generator/solver_pass_gen/axes_reorder_solver/axes_reorder_solver_gen.cpp	修改	生成双视图、映射和完整运行时选择代码
autofuse/att/generator/solver_pass_gen/axes_reorder_solver/axes_reorder_solver_gen.h	修改	增加 order、变量关系及 preferred 投影接口
autofuse/att/generator/solver_pass_gen/solver_pass_manager.cpp	修改	向 codegen 传递 canonical order 和变量关系
autofuse/att/generator/solver_pass/axes_reorder_solver_code/axes_reorder_solver_data_struct.cpp	修改	AxesReorderSolverInput 增加 ordered 视图和映射
autofuse/att/generator/solver_pass/axes_reorder_solver_code/axes_reorder_solver_equal_priority.cpp	修改	跳过单轴组并在 canonical 上识别双轴组
autofuse/att/generator/solver_pass/axes_reorder_solver_code/axes_reorder_solver_local_buf.cpp	修改	普通 solver 和 WorkloadBalance 使用 ordered 遍历
autofuse/att/generator/solver_pass/axes_reorder_solver_code/axes_reorder_solver_autotuning.cpp	修改	PGO ordered 枚举和 canonical 候选回填
autofuse/att/generator/solver_pass/axes_reorder_solver_code/axes_reorder_solver_main.cpp	修改	初始化并展示 ordered 视图
autofuse/tests/ut/att/testcase/gen_model_info/expr_gen/test_arg_list_reorder.cpp	修改	补充规则生成和 equal-order 稳定性 UT
autofuse/tests/ut/att/testcase/preprocess/test_args_manager.cpp	修改	补充缺失与歧义 order fallback UT
autofuse/tests/ut/att/testcase/solver_pass_gen/axes_reorder_gen/test_axes_reorder_gen.cpp	修改	补充普通、PGO、WorkloadBalance 代码生成 UT