| 文件 | 最后提交记录 | 最后更新时间 |
|---|---|---|
add kirin9030 ops: index matmul pooling Co-authored-by: zengjuan<zengjuan2@huawei.com> # message auto-generated for no-merge-commit merge: !1091 merge master into master add kirin9030 ops: index matmul pooling Created-by: zengjuan Commit-by: zengjuan Merged-by: cann-robot Description: ## 描述 add kirin9030 ops: index matmul pooling index/apply_top_k_top_p_with_sorted index/embedding_bag index/gather_elements_v2 index/inplace_index_add_with_sorted index/linear_index index/scatter_add_with_sorted index/scatter_list matmul/batch_mat_mul_v3 matmul/gemm_v2 matmul/mat_mul_v3 matmul/quant_batch_matmul_v3 matmul/quant_batch_matmul_v4 matmul/transpose_batch_mat_mul matmul/weight_quant_batch_matmul_v2 pooling/adaptive_avg_pool3d pooling/adaptive_max_pool3d pooling/avg_pool3_d pooling/max_pool3d_with_argmax_v2 ## 关联的Issue https://gitcode.com/cann/ops-nn/issues/723 ## 测试 蓝区门禁、黄区门禁、算子二级冒烟 ## 文档更新 无 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-nn!1091 | 6 个月前 | |
修复GatherElementsV2非尾轴场景同步错误 Co-authored-by: huang-wei-chen<chenhuangwei1@huawei.com> # message auto-generated for no-merge-commit merge: !2828 merge gather_elements_v2 into master 修复GatherElementsV2非尾轴场景同步错误 Created-by: huang-wei-chen Commit-by: huang-wei-chen Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> 在GatherProcess增加MTE3到Vector的同步，确保多轮计算数据正确 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> 关联Issue [#1542](https://gitcode.com/cann/ops-nn/issues/1542) ## 测试 <!--描述进行了哪些测试来验证你的改动。--> 算子泛化1890个用例通过 典型case复现脚本，修改后输出为True python import torch, torch_npu x = torch.randn(8192, 128, dtype=torch.float32) index = torch.randint(0, 8192, size=(8192, 8,)).to(torch.int32) out_npu = torch.gather(x.npu(), 0, index.npu()) out_cpu = torch.gather(x, 0, index.to(torch.int64)) print(f"结果是否一致：{torch.equal(out_cpu, out_npu.cpu())}") ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> NA ## 类型标签 <!-- [x] 表示选中 --> - [x] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-nn!2828 | 5 个月前 | |
代码同步 Co-authored-by: yang-di52<yangdi52@huawei.com> # message auto-generated for no-merge-commit merge: !450 merge master into master 代码同步 Created-by: yang-di52 Commit-by: yang-di52 Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-nn!450 | 8 个月前 | |
代码同步 Co-authored-by: yang-di52<yangdi52@huawei.com> # message auto-generated for no-merge-commit merge: !450 merge master into master 代码同步 Created-by: yang-di52 Commit-by: yang-di52 Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-nn!450 | 8 个月前 | |
修复GatherElementsV2非尾轴场景同步错误 Co-authored-by: huang-wei-chen<chenhuangwei1@huawei.com> # message auto-generated for no-merge-commit merge: !2828 merge gather_elements_v2 into master 修复GatherElementsV2非尾轴场景同步错误 Created-by: huang-wei-chen Commit-by: huang-wei-chen Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> 在GatherProcess增加MTE3到Vector的同步，确保多轮计算数据正确 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> 关联Issue [#1542](https://gitcode.com/cann/ops-nn/issues/1542) ## 测试 <!--描述进行了哪些测试来验证你的改动。--> 算子泛化1890个用例通过 典型case复现脚本，修改后输出为True python import torch, torch_npu x = torch.randn(8192, 128, dtype=torch.float32) index = torch.randint(0, 8192, size=(8192, 8,)).to(torch.int32) out_npu = torch.gather(x.npu(), 0, index.npu()) out_cpu = torch.gather(x, 0, index.to(torch.int64)) print(f"结果是否一致：{torch.equal(out_cpu, out_npu.cpu())}") ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> NA ## 类型标签 <!-- [x] 表示选中 --> - [x] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-nn!2828 | 5 个月前 |
| 文件 | 最后提交记录 | 最后更新时间 |
|---|---|---|
| 6 个月前 | ||
| 5 个月前 | ||
| 8 个月前 | ||
| 8 个月前 | ||
| 5 个月前 |