| 文件 | 最后提交记录 | 最后更新时间 |
|---|---|---|
ops-nn仓添加算子及aclnn标杆函数 Co-authored-by: dengguojie<dengguojie@huawei.com> # message auto-generated for no-merge-commit merge: !2376 merge master into master ops-nn仓添加算子及aclnn标杆函数 Created-by: dengguojie Commit-by: dengguojie Merged-by: cann-robot Description: ## 描述 ops-nn仓添加算子及aclnn标杆函数 ## 关联的Issue https://gitcode.com/cann/ops-nn/issues/1808 ## 测试 <!--描述进行了哪些测试来验证你的改动。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-nn!2376 | 4 个月前 | |
add_cases Co-authored-by: zhaogrug<zhaohai16@h-partners.com> # message auto-generated for no-merge-commit merge: !7583 merge add_cases into master add_cases Created-by: zhaogrug Commit-by: zhaogrug Merged-by: cann-robot Description: ## 描述 新增仓内新版ttk用例 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> ## 关联的Issue [#4223](https://gitcode.com/cann/ops-nn/issues/4223) <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> ## 测试 冒烟测试，本地验证 <!--描述进行了哪些测试来验证你的改动。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [x] 其他，请描述：用例添加 ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [ ] AI辅助编写 See merge request: cann/ops-nn!7583 | 11 天前 | |
[Fix] Fix softmax family AR full_load binaryTmpLocalTensor buffer overflow Co-authored-by: zhuzixian-lr<zhuzixian1@h-partners.com> # message auto-generated for no-merge-commit merge: !7630 merge fix/softmax-grad-ar-binary-tmp-buffer into master [Fix] Fix softmax family AR full_load binaryTmpLocalTensor buffer overflow Created-by: zhuzixian-lr Commit-by: zhuzixian-lr Merged-by: cann-robot Description: ## 问题 softmax_grad、log_softmax_grad、confusion_softmax_grad 的 AR full_load tiling 中 BINARY_TMP_LOCAL_SHAPE = 512 写死，对应 kernel 中 AR_FULL_LOAD_BINARY_TMP_BYTES = 512。 当 r > 2*VL_FP32 时，buffer 实际需求为 ubFactor * outerLoopDstStride * sizeof(float)，与 r 和 ubFactor 相关，512 字节远不够（最严重场景溢出 6.9 倍）。 ## 修改 1. **softmax_grad_tiling.h**: SoftmaxGradARTilingData 新增 binaryTmpSize 字段（影响 softmax_grad、log_softmax_grad、confusion_softmax_grad 三个算子） 2. **softmax_grad_tiling_ar_full_load.cpp**: 删除写死512，新增 FindNearestPower2，动态计算 binaryTmpPerRow，将 binaryTmpPerRow 放入 ubFactor 分母 3. **softmax_grad/log_softmax_grad/confusion_softmax_grad kernel**: 删除写死常量，使用 tl_->binaryTmpSize 4. **log_softmax_v2 kernel**: 删除写死常量，使用 tl_->binaryTmpSize（tiling 已由 #7550 修复） 5. 同步修改所有 UT 文件（op_host tiling data size、op_kernel tiling_def、test case） ## 验证 - 编译4个算子出包安装成功 - TTK测试13用例全部 MEM=PASS，TilingKey=1000（AR full_load） - torch_npu精度验证：softmax_grad/log_softmax_v2/log_softmax_grad 全部 PASS - op_host UT 20 tests PASSED - op_kernel UT 18 tests PASSED 关联PR: #7550 See merge request: cann/ops-nn!7630 | 3 天前 | |
Co-authored-by: Lizexin312<lizexin2@huawei.com> # message auto-generated for no-merge-commit merge: !1351 merge lzx_logsoftmax into master LogSoftmaxGrad算子支持Ascend950实现 Created-by: Lizexin312 Commit-by: Lizexin312 Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> LogSoftmaxGrad算子支持Ascend950实现 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> [#910](https://gitcode.com/cann/ops-nn/issues/910) ## 测试 <!--描述进行了哪些测试来验证你的改动。--> ttk验证/本地example验证 ## 测试用例 std::vector<int64_t> gradOutputShape = {4, 2}; std::vector<int64_t> outputShape = {4, 2}; std::vector<int64_t> outShape = {4, 2}; void* gradOutputDeviceAddr = nullptr; void* outputDeviceAddr = nullptr; void* outDeviceAddr = nullptr; aclTensor* gradOutput = nullptr; aclTensor* output = nullptr; aclTensor* out = nullptr; std::vector<float> gradOutputHostData = {0, 1, 2, 3, 4, 5, 6, 7}; std::vector<float> outputHostData = {1, 1, 1, 2, 2, 2, 3, 3}; std::vector<float> outHostData = {0, 0, 0, 0, 0, 0, 0, 0}; ## 测试结果 result[0] is: -32.619381 result[1] is: -42.492508 result[2] is: -30.619381 result[3] is: -115.224899 result[4] is: -84.6686 result[5] is: -113.224899 result[6] is: -235.026443 result[7] is: -314.368591 ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-nn!1351 | 5 个月前 |