| 补充TransposedBatchedMatmul Co-authored-by: init__zhb__<zhanghaobo6@huawei.com> # message auto-generated for no-merge-commit merge: !882 merge br.bmm.transposed into master 补充TransposedBatchedMatmul Created-by: init__zhb__ Commit-by: init__zhb__ Merged-by: cann-robot Description: ## 描述 - 新增TransposedBatchedMatmul(A可以形如(M, B, K)或(K, B, M), B可以形如(N, B, K)或者(K, B, N)) - 将strided_batched_matmul_tla 和 strided_batched_matmul_transpose_tla 中的bmm的golden计算移到公共组件部分 其他: - 移除experimental目录下与78_matrix_inverse相关的测试件tests/optest/tests/test_78_matrix_inverse.py ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue,请在这里提供Issue链接。--> ## 原因 <!--说明此次改动的目的、解决的问题等,应与类型标签匹配 --> ## 测试 ### 单算子测试 已进行200例子泛化验证 bash $ ./output/bin/45_strided_transposed_batched_matmul_tla 24 67 283 577 0 Compare success. ... ### optest接入验证 测试通过(包含transposed以及非trans在Row/Col组合下的情况) bash $ pytest tests/test_45_strided_batched_matmul_tla.py ... collected 8 items tests/test_45_strided_batched_matmul_tla.py ........ [100%] ========================================================================= 8 passed in 33.18s ========================================================================= ### 性能测试 基本持平 batch M N K STATUS1 TASK_TIME1 STATUS2 TASK_TIME2 STATUS3 TASK_TIME3 TASK_TIME_AVG CATLASS_TASK_TIME SpeedUP 80 512 80 640 unknown 46.834999 unknown 48.091 unknown 47.761002 47.562334 47.277333 1.00602828 80 512 27 640 unknown 42.473 unknown 42.888 unknown 42.831001 42.730667 39.800667 1.073616857 ## 文档更新 <!--如果这个PR包含文档的更新,请在这里指出。例如:更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他,请描述: See merge request: cann/catlass!882 | 1 天前 |