| Files | Last commit | Last update |
|---|---|---|
MatrixDiag README更新 Co-authored-by: pengyiming7<pengyiming7@h-partners.com> # message auto-generated for no-merge-commit merge: !2140 merge readme into master MatrixDiag README更新 Created-by: pengyiming7 Commit-by: pengyiming7 Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> MatrixDiag README更新 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> https://gitcode.com/cann/ops-math/issues/1139 ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!2140 | 6 months ago | |
[feat] 迁移开源TF插件--conversion: dynamic_partition, matrix_diag, matrix_diag_part, matrix_set_diag, matrix_set_diag_v2 Co-authored-by: sunchun<sunchun4@h-partners.com> # message auto-generated for no-merge-commit merge: !5571 merge tf_plugin/conversion-group1 into master [feat] 迁移开源TF插件--conversion: dynamic_partition, matrix_diag, matrix_diag_part, matrix_set_diag, matrix_set_diag_v2 Created-by: sunchun Commit-by: sunchun Merged-by: cann-robot Description: ## 描述 迁移开源TF插件--conversion: dynamic_partition, matrix_diag, matrix_diag_part, matrix_set_diag, matrix_set_diag_v2 ## 关联的Issue https://gitcode.com/cann/ops-math/issues/3168 ## 测试 二级冒烟 ## 文档更新 不涉及 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!5571 | 1 month ago | |
docs: 修正 matrix_diag 文件头注释与头文件保护注释 Co-authored-by: tangpingchuan<tangpingchuan@huawei.com> # message auto-generated for no-merge-commit merge: !6600 merge docs/comment-cleanup-20261022-05 into master docs: 修正 matrix_diag 文件头注释与头文件保护注释 Created-by: pingchuantang Commit-by: tangpingchuan Merged-by: cann-robot Description: ## 描述 修正 conversion/matrix_diag 下多处注释与实际代码不一致的问题，并同步修正一处 clang-format 格式问题： - conversion/matrix_diag/op_graph/matrix_diag_graph_infer.cpp：\file 注释由 matrix_diag_infershape.cpp 修正为 matrix_diag_graph_infer.cpp。 - conversion/matrix_diag/op_host/matrix_diag_tiling_arch35.h：#endif 尾注释由 OPS_BUILT_IN_OP_TILING_RUNTIME_MATRIX_DIAG_TILING_H_ 修正为实际保护宏 OPS_MATH_CONVERSION_MATRIX_DIAG_OP_HOST_MATRIX_DIAG_TILING_ARCH35_H_；同时按仓库 pre-commit 规则修正 MatrixDiagTiling 构造函数初始化列表后的空格格式。 - conversion/matrix_diag/op_kernel/arch35/matrix_diag_scatter_high_slice_batch.h：\file 注释由 matrix_diag_big_n.h 修正为 matrix_diag_scatter_high_slice_batch.h。 - conversion/matrix_diag/op_kernel/arch35/matrix_diag_scatter_high_slice_yUB.h：\file 注释由 matrix_diag_big_n.h 修正为 matrix_diag_scatter_high_slice_yUB.h。 说明：conversion/im2col/op_kernel/arch35/im2col_norm_NHWC.h 的 \file 注释未在本 PR 修复。该文件存在既有的大范围 clang-format 差异（约 169 行），应用会引入与本 PR 无关的大规模格式变更，已留待单独处理。 改动不涉及业务逻辑、接口行为或配置默认值，仅修正注释内容与格式。 ## 关联的Issue - https://gitcode.com/cann/ops-math/issues/3822 ## 测试 - [x] 本地完整 pre-commit 检查通过（含 clang-format、空白检查、文件末尾换行） - [x] 改动仅为注释与格式调整，不改变编译产物和算子逻辑，无需算子功能/精度测试 ## 文档更新 本次仅修正源码中的文件头注释与头文件保护宏尾注释，无独立文档文件更新。 ## 类型标签 - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!6600 | 20 days ago | |
docs: 修正 matrix_diag 文件头注释与头文件保护注释 Co-authored-by: tangpingchuan<tangpingchuan@huawei.com> # message auto-generated for no-merge-commit merge: !6600 merge docs/comment-cleanup-20261022-05 into master docs: 修正 matrix_diag 文件头注释与头文件保护注释 Created-by: pingchuantang Commit-by: tangpingchuan Merged-by: cann-robot Description: ## 描述 修正 conversion/matrix_diag 下多处注释与实际代码不一致的问题，并同步修正一处 clang-format 格式问题： - conversion/matrix_diag/op_graph/matrix_diag_graph_infer.cpp：\file 注释由 matrix_diag_infershape.cpp 修正为 matrix_diag_graph_infer.cpp。 - conversion/matrix_diag/op_host/matrix_diag_tiling_arch35.h：#endif 尾注释由 OPS_BUILT_IN_OP_TILING_RUNTIME_MATRIX_DIAG_TILING_H_ 修正为实际保护宏 OPS_MATH_CONVERSION_MATRIX_DIAG_OP_HOST_MATRIX_DIAG_TILING_ARCH35_H_；同时按仓库 pre-commit 规则修正 MatrixDiagTiling 构造函数初始化列表后的空格格式。 - conversion/matrix_diag/op_kernel/arch35/matrix_diag_scatter_high_slice_batch.h：\file 注释由 matrix_diag_big_n.h 修正为 matrix_diag_scatter_high_slice_batch.h。 - conversion/matrix_diag/op_kernel/arch35/matrix_diag_scatter_high_slice_yUB.h：\file 注释由 matrix_diag_big_n.h 修正为 matrix_diag_scatter_high_slice_yUB.h。 说明：conversion/im2col/op_kernel/arch35/im2col_norm_NHWC.h 的 \file 注释未在本 PR 修复。该文件存在既有的大范围 clang-format 差异（约 169 行），应用会引入与本 PR 无关的大规模格式变更，已留待单独处理。 改动不涉及业务逻辑、接口行为或配置默认值，仅修正注释内容与格式。 ## 关联的Issue - https://gitcode.com/cann/ops-math/issues/3822 ## 测试 - [x] 本地完整 pre-commit 检查通过（含 clang-format、空白检查、文件末尾换行） - [x] 改动仅为注释与格式调整，不改变编译产物和算子逻辑，无需算子功能/精度测试 ## 文档更新 本次仅修正源码中的文件头注释与头文件保护宏尾注释，无独立文档文件更新。 ## 类型标签 - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!6600 | 20 days ago | |
docs: 修正 matrix_diag 文件头注释与头文件保护注释 Co-authored-by: tangpingchuan<tangpingchuan@huawei.com> # message auto-generated for no-merge-commit merge: !6600 merge docs/comment-cleanup-20261022-05 into master docs: 修正 matrix_diag 文件头注释与头文件保护注释 Created-by: pingchuantang Commit-by: tangpingchuan Merged-by: cann-robot Description: ## 描述 修正 conversion/matrix_diag 下多处注释与实际代码不一致的问题，并同步修正一处 clang-format 格式问题： - conversion/matrix_diag/op_graph/matrix_diag_graph_infer.cpp：\file 注释由 matrix_diag_infershape.cpp 修正为 matrix_diag_graph_infer.cpp。 - conversion/matrix_diag/op_host/matrix_diag_tiling_arch35.h：#endif 尾注释由 OPS_BUILT_IN_OP_TILING_RUNTIME_MATRIX_DIAG_TILING_H_ 修正为实际保护宏 OPS_MATH_CONVERSION_MATRIX_DIAG_OP_HOST_MATRIX_DIAG_TILING_ARCH35_H_；同时按仓库 pre-commit 规则修正 MatrixDiagTiling 构造函数初始化列表后的空格格式。 - conversion/matrix_diag/op_kernel/arch35/matrix_diag_scatter_high_slice_batch.h：\file 注释由 matrix_diag_big_n.h 修正为 matrix_diag_scatter_high_slice_batch.h。 - conversion/matrix_diag/op_kernel/arch35/matrix_diag_scatter_high_slice_yUB.h：\file 注释由 matrix_diag_big_n.h 修正为 matrix_diag_scatter_high_slice_yUB.h。 说明：conversion/im2col/op_kernel/arch35/im2col_norm_NHWC.h 的 \file 注释未在本 PR 修复。该文件存在既有的大范围 clang-format 差异（约 169 行），应用会引入与本 PR 无关的大规模格式变更，已留待单独处理。 改动不涉及业务逻辑、接口行为或配置默认值，仅修正注释内容与格式。 ## 关联的Issue - https://gitcode.com/cann/ops-math/issues/3822 ## 测试 - [x] 本地完整 pre-commit 检查通过（含 clang-format、空白检查、文件末尾换行） - [x] 改动仅为注释与格式调整，不改变编译产物和算子逻辑，无需算子功能/精度测试 ## 文档更新 本次仅修正源码中的文件头注释与头文件保护宏尾注释，无独立文档文件更新。 ## 类型标签 - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!6600 | 20 days ago | |
diag等算子冗余workspace改为分配0 Co-authored-by: xufeng12121<1074805447@qq.com> # message auto-generated for no-merge-commit merge: !6418 merge new_ww into master diag等算子冗余workspace改为分配0 Created-by: xufeng12121 Commit-by: xufeng12121 Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> diag等算子冗余workspace改为分配0 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> https://gitcode.com/cann/ops-math/issues/3678 ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [x] 其他，请描述：代码整改 See merge request: cann/ops-math!6418 | 23 days ago | |
im2col、matrix_diag、roll、grouped_bias_add_grad、reduce_var、reduce_std_v2 六个算子新增 ascend960dt 平台支持 Co-authored-by: liaohuming<liaohuming@huawei.com> # message auto-generated for no-merge-commit merge: !5844 merge master into master im2col、matrix_diag、roll、grouped_bias_add_grad、reduce_var、reduce_std_v2 六个算子新增 ascend960dt 平台支持 Created-by: liaohuming Commit-by: liaohuming Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> 为 im2col、matrix_diag、roll、grouped_bias_add_grad、reduce_var、reduce_std_v2 六个算子新增 ascend960dt 平台支持，使其可在 ascend960dt（Ascend960DT_968B7）芯片上完成编译与调用。 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> https://gitcode.com/cann/ops-math/issues/3614 ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!5844 | 27 days ago | |
产品名变更 Co-authored-by: gitcode-chenjiao<chenjiao31@huawei.com> # message auto-generated for no-merge-commit merge: !5912 merge master into master 产品名变更 Created-by: gitcode-chenjiao Commit-by: gitcode-chenjiao Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> AI处理器对应的产品型号名称更新 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> [#3258](https://gitcode.com/cann/ops-math/issues/3258) ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ok ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> *.md ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!5912 | 1 month ago |
MatrixDiag
产品支持情况
| 产品 | 是否支持 |
|---|---|
| Ascend 950PR&950DT系列产品 | √ |
| Atlas A3系列产品 | √ |
| Atlas A2系列产品 | √ |
| Atlas 200I/500 A2推理产品 | × |
| Atlas推理系列产品 | √ |
| Atlas训练系列产品 | √ |
功能说明
- 算子功能：返回一个给定批量对角值的批量对角Tensor。
参数说明
| 参数名 | 输入/输出/属性 | 描述 | 数据类型 | 数据格式 |
|---|---|---|---|---|
| x | 输入 | 待进行MatrixDiag计算的入参，公式中的x。 | DOUBLE, FLOAT32, FLOAT16, BFLOAT16, COMPLEX32, COMPLEX64, COMPLEX128, INT8, UINT8, INT16, UINT16, INT32, UINT32, INT64, UINT64, QINT8, QUINT8, QINT16, QUINT16, QINT32, HIFLOAT8, FLOAT8_E5M2, FLOAT8_E4M3FN, BOOL. | ND |
| out | 输出 | 待进行MatrixDiag计算的出参，公式中的out。 | DOUBLE, FLOAT32, FLOAT16, BFLOAT16, COMPLEX32, COMPLEX64, COMPLEX128, INT8, UINT8, INT16, UINT16, INT32, UINT32, INT64, UINT64, QINT8, QUINT8, QINT16, QUINT16, QINT32, HIFLOAT8, FLOAT8_E5M2, FLOAT8_E4M3FN, BOOL. | ND |
- Atlas训练系列产品、Atlas推理系列产品：不支持BFLOAT16。
调用说明
| 调用方式 | 调用样例 | 说明 |
|---|---|---|
| 图模式调用 | test_geir_matrix_diag | 通过算子IR构图方式调用MatrixDiag算子。 |