catlass_gitcode1/examples/common · sunhao_hw/catlass_gitcode1

文件	最后提交记录	最后更新时间
golden	补充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 个月前
golden.hpp	matrix_inverse算子的golden代码 Co-authored-by: dx<2829025546@qq.com> # message auto-generated for no-merge-commit merge: !830 merge master into master matrix_inverse算子的golden代码 Created-by: weixin_63557130 Commit-by: dx Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> 这个文件实现了矩阵求逆（Matrix Inverse）算子，属于 Catlass::golden 命名空间下的 CPU 端黄金参考实现（用于验证 Ascend NPU 上算子结果的正确性）。 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> https://gitcode.com/cann/catlass/issues/331 ## 原因 <!--说明此次改动的目的、解决的问题等，应与类型标签匹配 --> ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/catlass!830	1 个月前
helper.hpp	用pre-commit刷新主要文件的格式 Co-authored-by: Yuan Tao<yuantao313@outlook.com> Co-authored-by: yuantao<taoyuan18@huawei.com> # message auto-generated for no-merge-commit merge: !874 merge precommit into master 用pre-commit刷新主要文件的格式 Created-by: yuantao_ Commit-by: yuantao;Yuan Tao Merged-by: cann-robot Description: ## 描述刷新主要文件的格式，并补充关键字检查项，保证pre-commit能够通过 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> ## 原因 <!--说明此次改动的目的、解决的问题等，应与类型标签匹配 --> ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/catlass!874	1 个月前
options.hpp	新增trmm Co-authored-by: dx<2829025546@qq.com> # message auto-generated for no-merge-commit merge: !845 merge master into master 新增trmm Created-by: weixin_63557130 Commit-by: dx Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> 新增 examples/74_trmm 样例，支持 TRMM 参数解析、NPU 执行和 CPU golden 精度校验。新增 TRMM kernel 封装，支持基于 CATLASS GEMM/TLA 组件完成三角矩阵乘计算。新增 torch_catlass.trmm Python API 和 optest 用例。支持 float32 输入输出。支持 side = 0 / 1，覆盖 left-side 和 right-side TRMM。支持 uplo = 0 / 1，覆盖下三角和上三角。支持 trans = 0 / 1，覆盖不转置和转置。支持 alpha 输出缩放。 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> https://gitcode.com/cann/catlass/issues/331 ## 原因 <!--说明此次改动的目的、解决的问题等，应与类型标签匹配 --> ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> optest 精度用例 test_74_trmm.py 通过，覆盖 16 个 TRMM 参数组合。 examples/74_trmm 指定 shape 自测通过，36/36 cases passed。 ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/catlass!845	1 个月前