Pull Request已成功合入, 合并人@CANN-robot
(感谢 gaoxin 的贡献)变更摘要
本 PR 为 V2 LoadStoreStride 性能模型中的 stride_used 增加 4096 的常量上限,避免 stride 成本计算超出模型适用范围。改动位于 autofuse/att/gen_model_info/api_perf_register/utils/api_perf_utils.cpp:新增常量 kMaxStrideUsed = 4096U,并在 LoadStoreStrideV2Func 中通过 af::sym::Min(stride_used, CreateExpr(kMaxStrideUsed)) 对 stride_used 进行截断,随后再乘以 block_count 与惩罚系数 k 参与成本计算。
主要改动
- 新增
kMaxStrideUsed常量: 在匿名命名空间中定义constexpr uint32_t kMaxStrideUsed = 4096U,作为 stride 使用的上限值。 - 对
stride_used施加上限: 在LoadStoreStrideV2Func中,基于LimitedStrideUpperBound计算得到stride_used后,追加stride_used = af::sym::Min(stride_used, CreateExpr(kMaxStrideUsed)),将其限制在 4096 以内。 - 影响成本计算链路: 截断后的
stride_used继续参与af::sym::Mul(k, af::sym::Mul(block_count, stride_used))的惩罚项计算,确保最终 stride 成本不超出性能模型的有效适用范围。


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
| module | lgtm status | approve status |
|---|---|---|
| repo-cann/graph-autofusion | ✅ yangyongqiang0606, zhang_shengjie, xchu42 (3/2) | ✅ yangyongqiang0606 (1/1) |
💡 Tip:
- Committer can comment
/approveor/lgtm- Commenting
/approveimplies both code review (lgtm) and intent to merge (approve)
CLA Signature Pass
gcw_V3YyYBt1, thanks for your pull request. All authors of the commits have signed the CLA. 👍


/lgtm


/approve


Pull Request
描述
为 V2 LoadStoreStride 性能模型中的 stride_used 增加 4096 的常量上限,避免 stride 成本计算超出模型适用范围。
变更类型
关联的Issue
无
如何测试
描述测试此变更的步骤和前提条件:
核对清单
其他信息