Star1245

Star1245

cann-robotmatrix_diag_part算子添加隔离宏

747427b5创建于 1 天前历史提交

文件	最后提交记录	最后更新时间
examples	长尾算子MatrixDiagPart新增950实现 Co-authored-by: surezz<zhangyuwei31@huawei.com> # message auto-generated for no-merge-commit merge: !4156 merge MatrixDiagPart into master 长尾算子MatrixDiagPart新增950实现 Created-by: surezz Commit-by: surezz Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> MatrixDiagPart算子新增950实现 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> https://gitcode.com/cann/ops-math/issues/2384 <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> 411条门槛用例、xrun geir -s、-d通路测试、xrun tensorflow通路测试，david冒烟、二级冒烟已过 ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> 新增MatrixDiagPart算子README文档 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!4156	12 天前
op_graph	matrix_diag_part算子添加隔离宏 Co-authored-by: 张煜炜<zhangyuwei31@huawei.com> # message auto-generated for no-merge-commit merge: !4578 merge matrix_reg_op_modify into master matrix_diag_part算子添加隔离宏 Created-by: surezz Commit-by: 张煜炜 Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> matrix_diag_part算子添加隔离宏，防止原型重复定义 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> https://gitcode.com/cann/ops-math/issues/2595 <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> david冒烟、二级冒烟 ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [x] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!4578	1 天前
op_host	长尾算子MatrixDiagPart新增950实现 Co-authored-by: surezz<zhangyuwei31@huawei.com> # message auto-generated for no-merge-commit merge: !4156 merge MatrixDiagPart into master 长尾算子MatrixDiagPart新增950实现 Created-by: surezz Commit-by: surezz Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> MatrixDiagPart算子新增950实现 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> https://gitcode.com/cann/ops-math/issues/2384 <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> 411条门槛用例、xrun geir -s、-d通路测试、xrun tensorflow通路测试，david冒烟、二级冒烟已过 ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> 新增MatrixDiagPart算子README文档 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!4156	12 天前
op_kernel	长尾算子MatrixDiagPart新增950实现 Co-authored-by: surezz<zhangyuwei31@huawei.com> # message auto-generated for no-merge-commit merge: !4156 merge MatrixDiagPart into master 长尾算子MatrixDiagPart新增950实现 Created-by: surezz Commit-by: surezz Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> MatrixDiagPart算子新增950实现 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> https://gitcode.com/cann/ops-math/issues/2384 <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> 411条门槛用例、xrun geir -s、-d通路测试、xrun tensorflow通路测试，david冒烟、二级冒烟已过 ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> 新增MatrixDiagPart算子README文档 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!4156	12 天前
tests	长尾算子MatrixDiagPart新增950实现 Co-authored-by: surezz<zhangyuwei31@huawei.com> # message auto-generated for no-merge-commit merge: !4156 merge MatrixDiagPart into master 长尾算子MatrixDiagPart新增950实现 Created-by: surezz Commit-by: surezz Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> MatrixDiagPart算子新增950实现 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> https://gitcode.com/cann/ops-math/issues/2384 <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> 411条门槛用例、xrun geir -s、-d通路测试、xrun tensorflow通路测试，david冒烟、二级冒烟已过 ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> 新增MatrixDiagPart算子README文档 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!4156	12 天前
CMakeLists.txt	长尾算子MatrixDiagPart新增950实现 Co-authored-by: surezz<zhangyuwei31@huawei.com> # message auto-generated for no-merge-commit merge: !4156 merge MatrixDiagPart into master 长尾算子MatrixDiagPart新增950实现 Created-by: surezz Commit-by: surezz Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> MatrixDiagPart算子新增950实现 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> https://gitcode.com/cann/ops-math/issues/2384 <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> 411条门槛用例、xrun geir -s、-d通路测试、xrun tensorflow通路测试，david冒烟、二级冒烟已过 ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> 新增MatrixDiagPart算子README文档 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!4156	12 天前
README.md	长尾算子MatrixDiagPart新增950实现 Co-authored-by: surezz<zhangyuwei31@huawei.com> # message auto-generated for no-merge-commit merge: !4156 merge MatrixDiagPart into master 长尾算子MatrixDiagPart新增950实现 Created-by: surezz Commit-by: surezz Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> MatrixDiagPart算子新增950实现 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> https://gitcode.com/cann/ops-math/issues/2384 <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> 411条门槛用例、xrun geir -s、-d通路测试、xrun tensorflow通路测试，david冒烟、二级冒烟已过 ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> 新增MatrixDiagPart算子README文档 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!4156	12 天前

MatrixDiagPart

产品支持情况

产品	是否支持
Ascend 950PR/Ascend 950DT	√
Atlas A3 训练系列产品/Atlas A3 推理系列产品	×
Atlas A2 训练系列产品/Atlas A2 推理系列产品	×
Atlas 200I/500 A2 推理产品	×
Atlas 推理系列产品	×
Atlas 训练系列产品	×

功能说明

算子功能：返回输入矩阵的最内层矩阵的主对角线（k=0）。

参数说明

参数名	输入/输出/属性	描述	数据类型	数据格式
x	输入	待计算主对角线的入参，最后两个维度为矩阵维度。	float16、float32、int32、int8、uint8	ND
y	输出	主对角线结果，与x同dtype。	float16、float32、int32、int8、uint8	ND

约束说明

输入x的维度数需大于等于2且小于等于8，最后两个维度表示矩阵。
输出y的维度数比输入x少1，形状为x.shape[:-2]+[min(M, N)]。
支持的数据类型：float16、float32、int32、int8、uint8。
数据格式仅支持ND。
仅支持连续输入，不支持非连续输入。
支持空tensor（M=0或N=0时，输出为空tensor）。
该算子为MatrixDiagPart V1版本，仅提取主对角线（k=0）。

调用说明

调用方式	样例代码	说明
图模式	test_geir_matrix_diag_part	通过算子IR构图方式调用MatrixDiagPart算子，参见算子调用完成编译和验证。