| 文件 | 最后提交记录 | 最后更新时间 |
|---|---|---|
新增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 个月前 | |
style: apply clang-format to all changed C++ files for CI compliance | 1 个月前 |