| 文件 | 最后提交记录 | 最后更新时间 |
|---|---|---|
fix: one_hot mix kernel 填off_value恰好整除UB块时漏写最后一整块 Co-authored-by: xieshengwei1024<xieshengwei3@huawei.com> # message auto-generated for no-merge-commit merge: !5784 merge fix_one_hot_mix_offvalue_tail into master fix: one_hot mix kernel 填off_value恰好整除UB块时漏写最后一整块 Created-by: xieshengwei1024 Commit-by: xieshengwei1024 Merged-by: cann-robot Description: ## 问题 OneHot mix kernel(tiling key 1001,WITH_UB 路径)在填充 off_value 时,当每核填充量恰好是 UB 块大小的整数倍(>1 块)时, tailOffValueCalNum_ = actual % offValueMaxNum = 0,InitOffValueAndCopyOut 只写入 loop-1 个整块 + 0 长度尾块,导致每核漏写最后一个整块(offValueMaxNum 个元素),输出残留未初始化数据(表现为非预期的 0)。 ## 复现条件 floor(输入总量 × depth / AIV核数) % (ubSize / sizeof(输出dtype)) == 0 且商 > 1。 例:Ascend950PR(ub_size=253952,块=48640 元素 @int32): - 输入 2660×2048 int32(每核 97280 = 2×48640):修复前输出 zero=2723840,修复后 0 - 输入 3990×2048 int32(每核 145920 = 3×48640):修复前 zero=2722507,修复后 0 - int64 输出块减半(24320 元素),同样命中 ## 修改 tailOffValueCalNum_ = actualLastCoreTotalSize - (offValueInitLoop_ - 1) * offValueMaxNum,整除时尾块按一整块写入,保证填充覆盖完整。 ## 测试 950PR 上触发用例全部修复,非整除用例回归通过。 相关 Issue:#3224(仅关联,本 MR 合入后不关闭该 issue) See merge request: cann/ops-math!5784 | 29 天前 | |
fix: 移除 lin_space/one_hot/range 不必要的workspace分配 Co-authored-by: yifangao<gaoyifan20@huawei.com> # message auto-generated for no-merge-commit merge: !5339 merge workspace-remove-2 into master fix: 移除 lin_space/one_hot/range 不必要的workspace分配 Created-by: yifangao Commit-by: yifangao Merged-by: cann-robot Description: ## 描述 移除 lin_space、one_hot、range 三个算子不必要的 workspace 分配: 1. **host 侧不再预留 workspace**:tiling 中 workspace 大小由固定 16MB(SYSTEM_WORKSPACE_SIZE/RESERVED_WORKSPACE)改为 DEFAULT_WORKSPACE_SIZE(0),三个算子统一使用同名常量。 2. **kernel 侧移除 workspace 解析**:删除 SetSysWorkspace/GetUserWorkspace 调用及空指针校验,workspace 直接传入 op.Init。 3. **UT 消除 magic number**:将 one_hot 的 DEFAULT_WORKSPACE_SIZE 上移至头文件 one_hot_tiling_arch35.h,UT 中 expectWorkspaces 使用 optiling::DEFAULT_WORKSPACE_SIZE 替代字面量,后续调整 workspace 大小无需逐个修改用例。 其余为代码格式化调整(单行化、对齐、补文件末尾换行),无功能影响。 ## 关联的Issue https://gitcode.com/cann/ops-math/issues/3216 ## 测试 obp冒烟:http://7.215.10.221/scheduler/alljobs?length=100&isalljobs=true&search=20260917_135556551 David冒烟:http://7.215.10.221/scheduler/alljobs?length=100&isalljobs=true&search=20260917_094314808 ## 文档更新 <!--如果这个PR包含文档的更新,请在这里指出。例如:更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他,请描述: See merge request: cann/ops-math!5339 | 29 天前 |
| 文件 | 最后提交记录 | 最后更新时间 |
|---|---|---|
| 29 天前 | ||
| 29 天前 |