Star0
0
Code
Code
Issues
Pull Requests
Actions
Discussion
Wiki
Analysis
Star0
0
cann-robotcann-robotdocs(assign): clarify dtype support by product
cabc8203Add on 2 days agoHistory
FilesLast commitLast update
examples
fix: 整改责任田算子日志质量（修复日志规范问题） Co-authored-by: StoneChan_<chensitong2@huawei.com> # message auto-generated for no-merge-commit merge: !5023 merge log-quality-rectify into master fix: 整改责任田算子日志质量（修复日志规范问题） Created-by: StoneChan_ Commit-by: StoneChan_ Merged-by: cann-robot Description: #### 一、问题描述 （必填） 依据 CANN 日志规范检查规则，对责任田算子的运行时日志进行质量整改，修复以下问题： - 级别错配：GE IR 示例（test_geir_*.cpp）失败信息误标 INFO（应 ERROR），存在故障被忽略风险 - 拼写错误：caculate/negtive/faild/mat 等 - 语法错误："is failed" 中式结构、"fail to get" 缺时态、缺助动词/介词、must same 等 - 变量越限日志缺正确范围（未给出期望值） - 中文标点（全角顿号）残留 - 示例直接 printf/std::cout 绕过统一日志接口；内部缩写 [XIR] 无法理解 - UT 测试中冗余 std::cout 状态输出 影响算子：cumsum/view_copy/addr/split/split_v/unpack/adds/concat/pack/as_strided/eye/dynamic_partition/assign/masked_fill/strided_slice_grad/triu/tril/tensor_move/asin/power/concat_d/concat_v2 #### 二、环境信息 （可选） 不涉及（纯日志文案整改，无运行时行为变化） #### 三、重现步骤 （可选） 执行算子 UT / 查看 logs/plog 中相关算子日志 #### 四、预期结果 （可选） 日志级别、文案符合日志规范，便于问题定位 💡 备注（选填）：排除 experimental 目录，仅整改主仓 关联 PR：https://gitcode.com/cann/ops-math/merge_requests/5023 --- 关联的Issue：#2845 See merge request: cann/ops-math!5023
2 months ago
framework
迁移开源TF插件--conversion Co-authored-by: nextyale<yaochen15@huawei.com> # message auto-generated for no-merge-commit merge: !4310 merge transfer-conversion into master 迁移开源TF插件--conversion Created-by: nextyale Commit-by: nextyale Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [x] 其他，请描述：迁移TF插件 See merge request: cann/ops-math!4310
2 months ago
op_graph
feat: 补齐11个算子InferDataType推导并对齐RT2.0 InferShape语义 Co-authored-by: esok11<yangsifa@huawei.com> # message auto-generated for no-merge-commit merge: !6869 merge f1029 into master feat: 补齐11个算子InferDataType推导并对齐RT2.0 InferShape语义 Created-by: esok11 Commit-by: esok11 Merged-by: cann-robot Description: ## 描述 面向 ops-math RT1.0 → RT2.0 迁移，为本仓 11 个算子补齐 op_graph 侧 InferDataType 推导，并将 op_host InferShape 的边界语义与 RT1.0 对齐（未知秩、非常量 shape、标量等），避免 C1 日落 rt1.0 后图模式推导回归。 ### 改动原因 1. 部分算子只有 InferShape、缺少 InferDataType 注册，图模式下无法完成输出 dtype 推导。 2. 部分 op_host InferShape 与 RT1.0 语义不一致，例如：AsStrided 在 size 非/未知时误用常量值计算甚至越界；Assign 取 value（输入1）而非 ref（输入0）；DynamicPartition 只有 InferShapeRange 无 InferShape；Pack/Unpack 将未知秩当已知维计算；ReciprocalGrad 误用 elewise 推导导致广播场景结果错误。 ### 改动方法 1) 新增 op_graph InferDataType（输出 dtype 透传输入 dtype） - 单输出、透传输入 0：Adds、Asin、Power、Assign、TensorMove、ViewCopy - 多输出、逐个透传：Pack、Unpack、DynamicPartition（输入 0 x）、AsStrided（输入 0 x）、StridedSliceGrad（输入 4 dy） - 均为 IMPL_OP(<Op>).InferDataType(...) 注册；op_graph 源码由 cmake/func.cmake 的 file(GLOB_RECURSE ... op_graph/*.cpp) 自动收集，无需改 CMakeLists。 2) op_host InferShape 对齐 - AsStrided：size 张量缺失、size 描述符含未知维/未知秩、GetValueToShape 取常量失败时，输出置未知秩并返回成功（对齐 rt1.0 SetOutputForRank）。 - Assign：输出 shape 改为取输入 0（ref），修复 value 为标量（numel=1）时输出 shape 错误。 - DynamicPartition：新增 InferShape4DynamicPartition（输入未知秩 → 各输出未知秩；输出秩 = x秩 - partitions秩 + 1，第 0 维为 -1，其余维取 x 对应维；partitions 秩大于 x 秩报错），并注册 InferShape + InferShapeRange + OutputShapeDependOnCompute({0})。 - Pack：未知秩输入按通配处理，取首个已知秩输入作为基准，全未知秩则输出未知秩；保留 N 与实例数一致性校验，失败日志改为同时打印基准索引。 - Unpack：输入未知秩时各输出保持未知秩，不再退化为 0 维。 - StridedSliceGrad：shape 输入形状值为 0（前向为 rank-0 标量）时输出 0 维标量，不再报错。 - ReciprocalGrad：InferShape4Elewise → InferShape4Broadcast。 - Adds/Asin/TensorMove/ViewCopy：仅 namespace 注释与格式修正，逻辑不变。 3) 测试 新增 39 个 host 侧 InferShape UT，覆盖未知秩/非常量/标量/广播/失败分支。 ## 关联的Issue - https://gitcode.com/cann/ops-math/issues/4383 ## 测试 - 新增 UT：as_strided 6 例、assign 2 例、concat 7 例、concat_d 4 例、pack 7 例、unpack 6 例、reciprocal_grad 5 例、strided_slice_grad 2 例（共 39 例，全部位于各算子 tests/ut/op_host）。 - 说明：本次流程未在本环境执行编译与 UT 运行，用例已随 PR 提交，需由 CI/本地环境验证。 ## 文档更新 无 ## 类型标签 - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!6869
12 days ago
op_host
feat: 补齐11个算子InferDataType推导并对齐RT2.0 InferShape语义 Co-authored-by: esok11<yangsifa@huawei.com> # message auto-generated for no-merge-commit merge: !6869 merge f1029 into master feat: 补齐11个算子InferDataType推导并对齐RT2.0 InferShape语义 Created-by: esok11 Commit-by: esok11 Merged-by: cann-robot Description: ## 描述 面向 ops-math RT1.0 → RT2.0 迁移，为本仓 11 个算子补齐 op_graph 侧 InferDataType 推导，并将 op_host InferShape 的边界语义与 RT1.0 对齐（未知秩、非常量 shape、标量等），避免 C1 日落 rt1.0 后图模式推导回归。 ### 改动原因 1. 部分算子只有 InferShape、缺少 InferDataType 注册，图模式下无法完成输出 dtype 推导。 2. 部分 op_host InferShape 与 RT1.0 语义不一致，例如：AsStrided 在 size 非/未知时误用常量值计算甚至越界；Assign 取 value（输入1）而非 ref（输入0）；DynamicPartition 只有 InferShapeRange 无 InferShape；Pack/Unpack 将未知秩当已知维计算；ReciprocalGrad 误用 elewise 推导导致广播场景结果错误。 ### 改动方法 1) 新增 op_graph InferDataType（输出 dtype 透传输入 dtype） - 单输出、透传输入 0：Adds、Asin、Power、Assign、TensorMove、ViewCopy - 多输出、逐个透传：Pack、Unpack、DynamicPartition（输入 0 x）、AsStrided（输入 0 x）、StridedSliceGrad（输入 4 dy） - 均为 IMPL_OP(<Op>).InferDataType(...) 注册；op_graph 源码由 cmake/func.cmake 的 file(GLOB_RECURSE ... op_graph/*.cpp) 自动收集，无需改 CMakeLists。 2) op_host InferShape 对齐 - AsStrided：size 张量缺失、size 描述符含未知维/未知秩、GetValueToShape 取常量失败时，输出置未知秩并返回成功（对齐 rt1.0 SetOutputForRank）。 - Assign：输出 shape 改为取输入 0（ref），修复 value 为标量（numel=1）时输出 shape 错误。 - DynamicPartition：新增 InferShape4DynamicPartition（输入未知秩 → 各输出未知秩；输出秩 = x秩 - partitions秩 + 1，第 0 维为 -1，其余维取 x 对应维；partitions 秩大于 x 秩报错），并注册 InferShape + InferShapeRange + OutputShapeDependOnCompute({0})。 - Pack：未知秩输入按通配处理，取首个已知秩输入作为基准，全未知秩则输出未知秩；保留 N 与实例数一致性校验，失败日志改为同时打印基准索引。 - Unpack：输入未知秩时各输出保持未知秩，不再退化为 0 维。 - StridedSliceGrad：shape 输入形状值为 0（前向为 rank-0 标量）时输出 0 维标量，不再报错。 - ReciprocalGrad：InferShape4Elewise → InferShape4Broadcast。 - Adds/Asin/TensorMove/ViewCopy：仅 namespace 注释与格式修正，逻辑不变。 3) 测试 新增 39 个 host 侧 InferShape UT，覆盖未知秩/非常量/标量/广播/失败分支。 ## 关联的Issue - https://gitcode.com/cann/ops-math/issues/4383 ## 测试 - 新增 UT：as_strided 6 例、assign 2 例、concat 7 例、concat_d 4 例、pack 7 例、unpack 6 例、reciprocal_grad 5 例、strided_slice_grad 2 例（共 39 例，全部位于各算子 tests/ut/op_host）。 - 说明：本次流程未在本环境执行编译与 UT 运行，用例已随 PR 提交，需由 CI/本地环境验证。 ## 文档更新 无 ## 类型标签 - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!6869
12 days ago
op_kernel
update operators Co-authored-by: biabu111<hebaojing1@huawei.com> # message auto-generated for no-merge-commit merge: !916 merge master into master update operators Created-by: biabu111 Commit-by: biabu111 Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> 增强以下算子功能 tensor_equal tensor_move tril triu stalessrandomuniform stalessrandomnormal assign dsarandomuniform dsarandomnormal ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> #530 ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!916
9 months ago
tests
fix: 整改责任田算子日志质量（修复日志规范问题） Co-authored-by: StoneChan_<chensitong2@huawei.com> # message auto-generated for no-merge-commit merge: !5023 merge log-quality-rectify into master fix: 整改责任田算子日志质量（修复日志规范问题） Created-by: StoneChan_ Commit-by: StoneChan_ Merged-by: cann-robot Description: #### 一、问题描述 （必填） 依据 CANN 日志规范检查规则，对责任田算子的运行时日志进行质量整改，修复以下问题： - 级别错配：GE IR 示例（test_geir_*.cpp）失败信息误标 INFO（应 ERROR），存在故障被忽略风险 - 拼写错误：caculate/negtive/faild/mat 等 - 语法错误："is failed" 中式结构、"fail to get" 缺时态、缺助动词/介词、must same 等 - 变量越限日志缺正确范围（未给出期望值） - 中文标点（全角顿号）残留 - 示例直接 printf/std::cout 绕过统一日志接口；内部缩写 [XIR] 无法理解 - UT 测试中冗余 std::cout 状态输出 影响算子：cumsum/view_copy/addr/split/split_v/unpack/adds/concat/pack/as_strided/eye/dynamic_partition/assign/masked_fill/strided_slice_grad/triu/tril/tensor_move/asin/power/concat_d/concat_v2 #### 二、环境信息 （可选） 不涉及（纯日志文案整改，无运行时行为变化） #### 三、重现步骤 （可选） 执行算子 UT / 查看 logs/plog 中相关算子日志 #### 四、预期结果 （可选） 日志级别、文案符合日志规范，便于问题定位 💡 备注（选填）：排除 experimental 目录，仅整改主仓 关联 PR：https://gitcode.com/cann/ops-math/merge_requests/5023 --- 关联的Issue：#2845 See merge request: cann/ops-math!5023
2 months ago
CMakeLists.txt
feat: 新增 ascend960pr/ascend960dt 平台算子适配（14 个算子） Co-authored-by: esok11<yangsifa@huawei.com> # message auto-generated for no-merge-commit merge: !6157 merge f1008 into master feat: 新增 ascend960pr/ascend960dt 平台算子适配（14 个算子） Created-by: esok11 Commit-by: esok11 Merged-by: cann-robot Description: ## 描述 为 ascend960pr / ascend960dt 两个新增芯片形态适配算子，覆盖 conversion 与 math 目录下共 14 个算子；同时修正 addr 的 dtype 支持列表判断逻辑，并对个别文件做版权头/格式整理。 ### 改动原因 ascend960pr / ascend960dt 为新增芯片形态，现有算子需补齐平台注册、编译单元与二进制配置后，才能在新平台上被算子信息库发现、编译并选择正确的 kernel 二进制。 ### 改动方法 1. **算子信息库注册**（*_def.cpp）：新增 this->AICore().AddConfig("ascend960pr", ...) 与 AddConfig("ascend960dt", ...) - conversion：assign、dynamic_partition、masked_fill、mem_set、tensor_move、tril、triu - math：addr、asin、eye、mod、sinc、truncate_div、truncate_mod 2. **编译配置**（CMakeLists.txt）：在 SUPPORT_COMPUTE_UNIT / COMPUTE_UNIT 中追加 ascend960pr、ascend960dt，并在 SUPPORT_TILING_DIR / TILING_DIR 中追加对应的 arch35（tiling 目录沿用 arch35） 3. **二进制配置**：为每个算子新增 op_host/config/ascend960pr/ 与 op_host/config/ascend960dt/ 目录，包含 <op>_binary.json 与 <op>_simplified_key.ini 4. **addr dtype 判断修正**（math/addr/op_api/aclnn_addr.cpp）：GetDtypeSupportList() 的判断由区间判断 >= ASCEND910B && <= ASCEND910E 改为显式判断 == ASCEND910B || == ASCEND910_93 || IsRegBase()，使 reg-base 平台复用 910B dtype 支持列表 5. **格式整理**：conversion/mem_set_v2/op_host/mem_set_v2_def.cpp、math/adds/op_host/adds_def.cpp、math/adds/CMakeLists.txt 仅做版权头/空行/行尾换行等整理 ## 关联的Issue - https://gitcode.com/cann/ops-math/issues/3440 ## 测试 本 PR 未包含测试代码改动；适配涉及算子信息库、编译配置与二进制配置，需在 ascend960pr / ascend960dt 平台补充二级冒烟与算子泛化验证。 ## 文档更新 无。 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!6157
1 month ago
README.md
docs(assign): clarify dtype support by product Co-authored-by: Dyrong<dengyirong@huawei.com> # message auto-generated for no-merge-commit merge: !7392 merge fix/assign-readme-double-4236 into master docs(assign): clarify dtype support by product Created-by: Dyrong Commit-by: Dyrong Merged-by: cann-robot Description: ## 描述 修改readme资料文档说明，补充DOUBLE类型和不同设备支持的数据类型说明 ## 关联的Issue [#4236](https://gitcode.com/cann/ops-math/issues/4236) ## 测试 ## 文档更新 README.md、README_en.md ## 类型标签 - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!7392
2 days ago
README_en.md
docs(assign): clarify dtype support by product Co-authored-by: Dyrong<dengyirong@huawei.com> # message auto-generated for no-merge-commit merge: !7392 merge fix/assign-readme-double-4236 into master docs(assign): clarify dtype support by product Created-by: Dyrong Commit-by: Dyrong Merged-by: cann-robot Description: ## 描述 修改readme资料文档说明，补充DOUBLE类型和不同设备支持的数据类型说明 ## 关联的Issue [#4236](https://gitcode.com/cann/ops-math/issues/4236) ## 测试 ## 文档更新 README.md、README_en.md ## 类型标签 - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!7392
2 days ago
README

Assign

English | 简体中文

Product Support Status

Product Supported
Ascend 950PR&950DT products √
Atlas A3 products √
Atlas A2 products √
Atlas 200I/500 A2 inference products ×
Atlas inference products √
Atlas training products √

Function Description

  • Operator function: fills the value of one input tensor into another input tensor.

  • Computation formula:

outi=valueiout_i=value_i

Parameter Description

Parameter Name Input/Output/Attribute Description Data Type Data Format
ref Input Input parameter for the Assign computation, which is out in the formula. The input is also the output. FLOAT, FLOAT16, BFLOAT16, DOUBLE, INT8, UINT8, INT16, UINT16, INT32, UINT32, INT64, UINT64, BOOL, COMPLEX32, COMPLEX64 ND
value Input Fill value for the Assign computation, which is value in the formula. FLOAT, FLOAT16, BFLOAT16, DOUBLE, INT8, UINT8, INT16, UINT16, INT32, UINT32, INT64, UINT64, BOOL, COMPLEX32, COMPLEX64 ND
ref Output Output parameter for the Assign computation, which is out in the formula. FLOAT, FLOAT16, BFLOAT16, DOUBLE, INT8, UINT8, INT16, UINT16, INT32, UINT32, INT64, UINT64, BOOL, COMPLEX32, COMPLEX64 ND

Constraint Description

  • Ascend 950PR&950DT products: The supported data types are FLOAT, FLOAT16, BFLOAT16, DOUBLE, INT8, UINT8, INT16, UINT16, INT32, UINT32, INT64, UINT64, BOOL, COMPLEX32, and COMPLEX64.
  • Atlas A3 products and Atlas A2 products: The supported data types are FLOAT, FLOAT16, BFLOAT16, INT8, UINT8, INT16, UINT16, INT32, UINT32, INT64, UINT64, and BOOL.
  • Atlas inference products and Atlas training products: The supported data types are FLOAT, FLOAT16, INT8, UINT8, INT16, UINT16, INT32, UINT32, INT64, UINT64, and BOOL.

Invocation Description

Invocation Method Invocation Sample Description
Graph-mode invocation test_geir_assign Invokes the Assign operator by building a graph through the operator IR.