Pull Request

变更摘要

本 PR 修复 CV UBFuse 场景下 host tiling 求解器与 device 侧 tmp 分配不一致导致的 UB 越界问题。由于 GetTmpBuffer 会为 tmp 公式套上 v1 repeat cap 壳（Min(size, 255 * 256 + 32)），而 CV UBFuse 的 Pow epilogue 内联公式按原始需求分配 tmp，两侧不一致使求解器误判 UB 可行并选出较大 stage，最终 buf_cube + b0_raw + 队列 超出 UB。改动将 is_cv_ub_fusion 透传给 AscGraphUbExprBuilder，在 CV UBFuse 下对 tmp 剥离 v1 repeat cap，使 UB 约束与 device 分配保持一致，从而让求解器收缩 stage 或走 common 兜底。

主要改动

• 新增 v1 repeat cap 剥离逻辑：在 asc_graph_ub_expr_builder.cpp 中新增常量 kMaxV1RepeatTmpBytes 与 StripV1RepeatCap，对形如 Min(size, 255 * 256 + 32) 的表达式剥离 cap 壳，还原原始 tmp 需求。
• Build 接口增加原始 tmp 尺寸开关：AscGraphUbExprBuilder::Build 与 AddTmpBuffer 新增 use_raw_tmp_size 参数，为真时按原始 tmp 需求累加 node_tmp_buffer_bytes，默认值为 false 保持原有行为。
• UbExprContext 新增字段：ub_expr_types.h 中 UbExprContext 增加 use_raw_tmp_size 字段，用于承载 CV UBFuse 下 UB 约束按原始 tmp 计算的语义。
• host 侧透传 CV UBFuse 标识：gen_model_info.cpp 中 RefreshCommonUbExprContext 新增 is_cv_ub_fusion 参数并传入 Build，使 GenerateSingleModelInfoWithContext 能把 CV UBFuse 状态传递到 UB 表达式构建。
• 补充单元测试：test_ub_expr_utils.cpp 新增 BuildStripsV1RepeatCapForRawTmpSize，验证默认构建得到 capped 值 65312、开启 use_raw_tmp_size 后得到原始值 80000。

代码审查

✅ 未发现问题

Thanks for your pull-request.
The full list of commands accepted by me can be found at here.
You can get sig-info at here.
You can self-configure the PR merge rules for this repository. For more details, please refer to Here.
For more, you also can visit HICANN.

PR Approval Progress

✅ Congratulations! All modules have met the lgtm and approve requirements.

Module Approval Details

💡 Tip:

• Committer can comment /approve or /lgtm
• Commenting /approve implies both code review (lgtm) and intent to merge (approve)

CLA Signature Pass

tianbaolin, thanks for your pull request. All authors of the commits have signed the CLA. 👍

compile

compile

/lgtm

/lgtm

/approve