| 文件 | 最后提交记录 | 最后更新时间 |
|---|---|---|
【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 | 6 个月前 | |
MatrixSetDiag修改proto描述 Co-authored-by: onanfield<tianshang@huawei.com> # message auto-generated for no-merge-commit merge: !1514 merge master into master MatrixSetDiag修改proto描述 Created-by: onanfield Commit-by: onanfield Merged-by: cann-robot Description: ## 描述 MatrixSetDiag修改proto描述 ## 关联的Issue https://gitcode.com/cann/ops-math/issues/884 ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 ## 类型标签 <!-- [x] 表示选中 --> - [x] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!1514 | 5 个月前 | |
fix: matrix_set_diag系列算子整改魔鬼数字、类型转换、除零及回绕风险 Co-authored-by: Xbying<2119046648@qq.com> # message auto-generated for no-merge-commit merge: !4788 merge 5 into master fix: matrix_set_diag系列算子整改魔鬼数字、类型转换、除零及回绕风险 Created-by: 2301_81474000 Commit-by: Xbying Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> 1. matrix_set_diag_v2_no_cut_tail.h — 消除魔鬼数字 3 处 2 字面量（processSize *= 2 ×2、tailSize * 2）复用已有常量 ELEMENT_EXPAND_TIMES，与同文件 421/532/533 行用法保持一致 2. matrix_set_diag_v2_tiling_arch35_base.cpp — 类型安全 + 除零防护 ratio 计算：(double)(...) C 风格转换 → static_cast<double> - CalculateUbFactorAndCheck：ubComputeBufSize / totalTailSize 前增加 totalTailSize == 0 判断，置 ubFactor_ = 1 返回，杜绝极小 shape 下除零 3. matrix_set_diag_v2_tiling_arch35.cpp — 回绕防护 + 成员初始化 - CheckDiag：GetDim(diagDimNum_ - 1) 前加 diagDimNum_ < 1 校验 - CheckK：GetDim(diagDimNum_ - 2) 前加 diagDimNum_ < 2 校验，防止 uint32 下溢回绕 - 成员变量：context_ = nullptr、inputDataType_ = ge::DT_UNDEFINED 显式初始化 - 防御写法与同族算子 matrix_set_diag_tiling_arch35.cpp:102 惯例一致 4. matrix_set_diag_tiling_arch35.cpp — 显式类型转换 - GetDim() 返回 int64_t 赋给 uint64_t maxDiagLen 增加 static_cast<uint64_t> 5. matrix_set_diag_v2_tiling_arch35_base.h — 形参命名 - FillsTilingData 形参 baseTilingData → tilingData，与其他重载风格对齐 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> https://gitcode.com/cann/ops-math/issues/2664 ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [X] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!4788 | 20 天前 | |
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 | 1 个月前 | |
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 | 24 天前 | |
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 | 1 个月前 | |
doc Tools扫描错误修改 Co-authored-by: gitee-yanglulu<yanglulul@h-partners.com> # message auto-generated for no-merge-commit merge: !1899 merge master into master doc Tools扫描错误修改 Created-by: gitee-yanglulu Commit-by: gitee-yanglulu Merged-by: cann-robot Description: doc Tools扫描错误修改 See merge request: cann/ops-math!1899 | 5 个月前 |
MatrixSetDiag
产品支持情况
| 产品 | 是否支持 |
|---|---|
| Ascend 950PR/Ascend 950DT | √ |
| Atlas A3 训练系列产品/Atlas A3 推理系列产品 | × |
| Atlas A2 训练系列产品/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算子。 |