| Files | Last commit | Last update |
|---|---|---|
【MatrixSetDiag】 Co-authored-by: chenyuqiao<chenyuqiao5@huawei.com> # message auto-generated for no-merge-commit merge: !1270 merge msd_0213 into master 【MatrixSetDiag】 Created-by: chenyuqiao Commit-by: chenyuqiao Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> - feat：新增 MatrixSetDiag ascend950实现，包括simd/simt等模板 - perf：重构优化 infershape 代码 - feat：新增 MatrixSetDiag geir example ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> [Requirement|需求建议]: 支持MatrixSetDiag算子AscendC实现 ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> - conversion\matrix_set_diag\README.md ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!1270 | 8 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 | |
日志整改(im2col等算子) Co-authored-by: wow5523<wuao1@huawei.com> # message auto-generated for no-merge-commit merge: !5009 merge log-remediation into master 日志整改(im2col等算子) Created-by: zhu-xun Commit-by: wow5523 Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> 日志整改(im2col等算子) ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [x] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!5009 | 2 months ago | |
support ascend960dt/ascend960pr Co-authored-by: yangjinwen<yangjinwen2@huawei.com> # message auto-generated for no-merge-commit merge: !6523 merge dev-v121-bugfix into master support ascend960dt/ascend960pr Created-by: yangjinwen Commit-by: yangjinwen Merged-by: cann-robot Description: ## 描述 ascend960dt/ascend960pr 平台缺失部分算子的二进制编译配置，且 Concat→ConcatD 融合 pass 的 RegBase 平台判断未覆盖 ascend960dt（DAV_9201）架构，导致相关算子在 ascend960dt/ascend960pr 平台不可用。 本次改动： 1. conversion/concat_d/op_graph/fusion_pass/concat_to_concatd_fusion_pass.cpp：IsTargetPlatform() 的 regbaseArch 白名单新增 NpuArch::DAV_9201，使 Concat→ConcatD 融合 pass 在 ascend960dt 平台生效，日志同步更新。 2. scripts/kernel/binary_config/ascendc_config.json：为 Concat、ConcatD、ConcatV2、Pack、BatchToSpaceND、MatrixSetDiag、SpaceToDepth、DepthToSpace、Rint 共 9 个算子补充 ascend960dt 计算单元（compute_units）及对应编译选项配置。 3. **MatrixSetDiag 新增 ascend960dt/ascend960pr 平台支持**： - conversion/matrix_set_diag/CMakeLists.txt：COMPUTE_UNIT 新增 ascend960dt、ascend960pr，TILING_DIR 对应 arch35； - conversion/matrix_set_diag/op_host/matrix_set_diag_def.cpp：AICore 配置注册 ascend960dt、ascend960pr； - 新增 op_host/config/ascend960dt/matrix_set_diag_binary.json 与 op_host/config/ascend960pr/matrix_set_diag_binary.json（int8/int16/int32/int64 四种 dtype 二进制配置，DtypeByte 匹配）。 ## 关联的Issue https://gitcode.com/cann/ops-math/issues/3681 See merge request: cann/ops-math!6523 | 23 days ago | |
feat: 新增MatrixSetDiagV2 A5实现 Co-authored-by: onanfield<tianshang@huawei.com> Co-authored-by: zhanw_coding<zhanwang@huawei.com> Co-authored-by: yefeicoding<yefei25@huawei.com> # message auto-generated for no-merge-commit merge: !3961 merge msdv2 into master feat: 新增MatrixSetDiagV2 A5实现 Created-by: zhanw_coding Commit-by: zhanw_coding;yefeicoding;onanfield Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> - feat: 新增MatrixSetDiagV2 A5实现 - feat: MatrixSetDiag 被 MatrixSetDiagV2 收编，合并实现 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> https://gitcode.com/cann/ops-math/issues/2417 ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> - conversion/matrix_set_diag_v2/README.md ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!3961 | 3 months ago | |
feat: add matrix_set_diag_v2 UT Co-authored-by: zhanw_coding<zhanwang@huawei.com> # message auto-generated for no-merge-commit merge: !4608 merge fix-msd into master feat: add matrix_set_diag_v2 UT Created-by: zhanw_coding Commit-by: zhanw_coding Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> - feat: add matrix_set_diag_v2 UT - fix: matrix_set_diag codecheck - feat: add matrix_set_diag_v2 classify rule 新增 14 个用例（全部通过） - infershape（7 个）： 非 const k 三分支（diag 维过小/过大/合法）、diag 未知前导维 continue、x 未知前导维用 diag 填充、x/diag 未知 rank（{-2}）→ 输出 {-2}。 - tiling（7 个）： V1 切尾路径 3 个用例（主循环/提前返回/CalUbFactor if 分支）、SIMT ubSize 过小失败、GATHER ubFactor=0 回退 CutTail、NoCutTail 优化循环 break、SIMT 直达。 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> https://gitcode.com/cann/ops-math/issues/2606 ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> UT ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [x] 其他，请描述：补充UT See merge request: cann/ops-math!4608 | 2 months ago | |
support ascend960dt/ascend960pr Co-authored-by: yangjinwen<yangjinwen2@huawei.com> # message auto-generated for no-merge-commit merge: !6523 merge dev-v121-bugfix into master support ascend960dt/ascend960pr Created-by: yangjinwen Commit-by: yangjinwen Merged-by: cann-robot Description: ## 描述 ascend960dt/ascend960pr 平台缺失部分算子的二进制编译配置，且 Concat→ConcatD 融合 pass 的 RegBase 平台判断未覆盖 ascend960dt（DAV_9201）架构，导致相关算子在 ascend960dt/ascend960pr 平台不可用。 本次改动： 1. conversion/concat_d/op_graph/fusion_pass/concat_to_concatd_fusion_pass.cpp：IsTargetPlatform() 的 regbaseArch 白名单新增 NpuArch::DAV_9201，使 Concat→ConcatD 融合 pass 在 ascend960dt 平台生效，日志同步更新。 2. scripts/kernel/binary_config/ascendc_config.json：为 Concat、ConcatD、ConcatV2、Pack、BatchToSpaceND、MatrixSetDiag、SpaceToDepth、DepthToSpace、Rint 共 9 个算子补充 ascend960dt 计算单元（compute_units）及对应编译选项配置。 3. **MatrixSetDiag 新增 ascend960dt/ascend960pr 平台支持**： - conversion/matrix_set_diag/CMakeLists.txt：COMPUTE_UNIT 新增 ascend960dt、ascend960pr，TILING_DIR 对应 arch35； - conversion/matrix_set_diag/op_host/matrix_set_diag_def.cpp：AICore 配置注册 ascend960dt、ascend960pr； - 新增 op_host/config/ascend960dt/matrix_set_diag_binary.json 与 op_host/config/ascend960pr/matrix_set_diag_binary.json（int8/int16/int32/int64 四种 dtype 二进制配置，DtypeByte 匹配）。 ## 关联的Issue https://gitcode.com/cann/ops-math/issues/3681 See merge request: cann/ops-math!6523 | 23 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 |
MatrixSetDiag
产品支持情况
| 产品 | 是否支持 |
|---|---|
| Ascend 950PR&950DT系列产品 | √ |
| Atlas A3系列产品 | × |
| Atlas A2系列产品 | × |
| Atlas 200I/500 A2推理产品 | × |
| Atlas推理系列产品 | × |
| Atlas训练系列产品 | × |
功能说明
- 算子功能：将输入tensor的对角线元素替换为对角线tensor的值。
参数说明
| 参数名 | 输入/输出/属性 | 描述 | 数据类型 | 数据格式 |
|---|---|---|---|---|
| x | 输入 | 待进行替换的原始tensor。 | FLOAT、FLOAT16、BFLOAT16、INT8、UINT8、INT16、UINT16、INT32、UINT32、INT64、UINT64、BOOL、COMPLEX64、DOUBLE | ND |
| diagonal | 输入 | 对角线tensor。 | 与x的数据类型保持一致。 | ND |
| y | 输出 | 进行替换后的tensor。 | 与x的数据类型保持一致。 | ND |
约束说明
- x的维度最少为2维，最大不超过8维。
- diagonal的维度比x的维度小1维，且最后一维的值为x最后两维的较小值。
- diagonal的维度除最后一维外其他维度要和x的维度一一对应相等。
调用说明
| 调用方式 | 调用样例 | 说明 |
|---|---|---|
| 图模式调用 | test_geir_matrix_set_diag | 通过算子IR构图方式调用MatrixSetDiag算子。 |