master

cann-robotfeat: 补齐11个算子InferDataType推导并对齐RT2.0 InferShape语义

3550c7f0Add on 8 days agoHistory

Files	Last commit	Last update
arch35	feat: viewcopy/as_strided/tensor_move 回合ascend960dt平台适配代码 Co-authored-by: yefeicoding<yefei25@huawei.com> # message auto-generated for no-merge-commit merge: !6191 merge v121_10 into master feat: viewcopy/as_strided/tensor_move 回合ascend960dt平台适配代码 Created-by: yefeicoding Commit-by: yefeicoding Merged-by: cann-robot Description: ## 描述 viewcopy/as_strided/tensor_move 回合ascend960dt平台适配代码 ## 关联的Issue #3498 <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!6191	25 days ago
config	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
tensor_move_def.cpp	feat: viewcopy/as_strided/tensor_move 回合ascend960dt平台适配代码 Co-authored-by: yefeicoding<yefei25@huawei.com> # message auto-generated for no-merge-commit merge: !6191 merge v121_10 into master feat: viewcopy/as_strided/tensor_move 回合ascend960dt平台适配代码 Created-by: yefeicoding Commit-by: yefeicoding Merged-by: cann-robot Description: ## 描述 viewcopy/as_strided/tensor_move 回合ascend960dt平台适配代码 ## 关联的Issue #3498 <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!6191	25 days ago
tensor_move_infershape.cpp	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	8 days ago