| Files | Last commit | Last update |
|---|---|---|
[CANNBOT]acosh_grad支持数据类型校验，population_count、scale golden修改 Co-authored-by: gcw_YBIAEfqJ<wangweidong15@huawei.com> # message auto-generated for no-merge-commit merge: !4381 merge ticket into master [CANNBOT]acosh_grad支持数据类型校验，population_count、scale golden修改 Created-by: gcw_YBIAEfqJ Commit-by: gcw_YBIAEfqJ Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> acosh_grad支持数据类型校验，population_count、scale golden修改 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> https://gitcode.com/cann/ops-math/issues/2478 ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [x] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!4381 | 3 months ago | |
整改st目录下的atk用例统一目录格式 Co-authored-by: yue-ma<mayue54@huawei.com> # message auto-generated for no-merge-commit merge: !2579 merge master into master 整改st目录下的atk用例统一目录格式 Created-by: yue-ma Commit-by: yue-ma Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> 整改st目录下的atk用例统一目录格式 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> https://gitcode.com/cann/ops-math/issues/1538 ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [x] 其他，请描述：st用例目录整改 See merge request: cann/ops-math!2579 | 5 months ago | |
fix(scale): arch35 kernel NDDMA 维度按 run 合并，消除大 shape 下 tiny DataCopy 性能瓶颈 Co-authored-by: babeiee<125106023003@njust.edu.cn> # message auto-generated for no-merge-commit merge: !5888 merge fix/scale-arch35-nddma-run-coalesce into master fix(scale): arch35 kernel NDDMA 维度按 run 合并，消除大 shape 下 tiny DataCopy 性能瓶颈 Created-by: babeiee Commit-by: babeiee Merged-by: cann-robot Description: ## 问题 Scale 算子在大 shape（如 x=(82545,2,2,2,2,2)、(102678,2,2,2,2,2,2,2) 等 rank6~8、外侧大维度 + 内侧小维度场景）性能极差：实测带宽仅 8~16 GB/s（对照单维大 shape 约 1070 GB/s）。 ## 根因 1. FindSplitAxis 只按 UB 容量从内向外选切分轴，未考虑 NDDMA 5 维上限。当最内 5 维连乘很小（如全 2，仅 32 元素）时，切分轴被迫落在最外层（RANK−k > 5），落入 CopyInBrc 的 outer loop 区。 2. 旧 CopyInBrc outer loop 每个外层迭代发一次 NDDMA DataCopy，每次仅搬最内 5 维（32 元素/64B），大 shape 下 DataCopy 次数达 10^5~10^6 次。 3. 加重因素：nddmaOuterIters_ 按整段 a_i 而非尾段 a_i_tail 计算，尾 tile 多搬约 3 倍废数据。 ## 修复（仅 kernel 侧，Host/TilingData 无改动） 将 tile 维度区间 [k, RANK) 按**可仿射合并**条件 strides[e] == dstShape[e+1] * strides[e+1] 划分为 run（broadcast 段 stride=0 自然可并，稠密连续段同理），每个 run 填一个 NDDMA 维： - x 恒为稠密输入 → 6~8 维合并为 1 个 run → 每 tile 单次 DataCopy 整块搬运； - scale/bias 合并为少量 run，单次 DataCopy 完成随路广播； - run 数超 5 维的病态输入保留 outer loop 兜底，外层次数按 a_i_seg 修正（顺带修复尾段多搬）； - 寻址按 NDDMA 线性模型推导，与旧逐维模型**逐元素等价**，UB 内数据排布不变，compute/CopyOut 零改动。 ## 验证 1. **寻址交叉验证**：Python 逐元素比对合并后 NDDMA 寻址与参考寻址，210 例（含 200 随机 fuzz、9 个目标样例结构、run 溢出兜底）全部一致； 2. **UT**（dav_3510 仿真）：新增 2 个 rank8 回归用例（切分轴在 NDDMA 窗口外含尾段 / run 溢出兜底），4/4 通过；Host tiling UT 16/16 通过； 3. **实卡性能**（Ascend950PR，30 次迭代均值，精度全量比对 PASS）： | 样例 | 修复前 | 修复后 | 加速比 | 带宽 |---|---|---|---|---| | L0_067/461 fp16 | 1.314 ms | 0.030 ms | 43.8× | 8.3→358 GB/s | | L1_137 fp32 | 10.829 ms | 0.185 ms | 58.5× | 15→895 GB/s | | L1_186 fp32 | 5.190 ms | 0.075 ms | 69.2× | 15→1058 GB/s | | L1_200 bf16 | 6.504 ms | 0.108 ms | 60.2× | 8→500 GB/s | | L1_390 fp32 | 9.531 ms | 0.148 ms | 64.4× | 16→1057 GB/s | | L1_406 fp16 | 8.087 ms | 0.127 ms | 63.7× | 12→755 GB/s | | L1_491 fp32 | 3.039 ms | 0.052 ms | 58.4× | 16→959 GB/s | | rangebound rb_748 bf16 | 5.206 ms | 0.073 ms | 71.3× | 8→578 GB/s | | 对照组（单维大 shape） | 0.589 ms | 0.590 ms | 1.0×（无回退） | — | See merge request: cann/ops-math!5888 | 1 month ago | |
增加A2 A3新算子支持 Co-authored-by: slx2008<shaolixin@huawei.com> # message auto-generated for no-merge-commit merge: !448 merge master into master 增加A2 A3新算子支持 Created-by: songkai111 Commit-by: slx2008 Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!448 | 10 months ago |