master

Files	Last commit	Last update
arch35	concat算子tilingdata压缩，降低头开销 Co-authored-by: hw-zhangpanpan<zhangpanpan5@huawei.com> # message auto-generated for no-merge-commit merge: !4196 merge master into master concat算子tilingdata压缩，降低头开销 Created-by: hw-zhangpanpan Commit-by: hw-zhangpanpan Merged-by: cann-robot Description: ## 描述 目前concat算子tilingdata达到1300+B,为了降低小case场景头开销，提升性能，先压缩减小tilingdata。 ## 关联的Issue https://gitcode.com/cann/ops-math/issues/2849 ## 测试 泛化用例验证 ## 文档更新 不涉及 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!4196	2 months ago
config	[ascend960dt/ascend960pr] conversion/math类9个算子新增ascend960dt/ascend960pr平台支持 Co-authored-by: yangjinwen<yangjinwen2@huawei.com> # message auto-generated for no-merge-commit merge: !5778 merge dev-v121 into master [ascend960dt/ascend960pr] conversion/math类9个算子新增ascend960dt/ascend960pr平台支持 Created-by: yangjinwen Commit-by: yangjinwen Merged-by: cann-robot Description: ## 描述 conversion 类 7 个算子（batch_to_space_nd、concat、concat_d、concat_v2、depth_to_space、pack、space_to_depth）与 math 类 2 个算子（erfc、rint）新增 ascend960dt / ascend960pr 平台支持，复用 arch35 tiling 框架，主要改动： - 平台注册：9 个算子 CMakeLists.txt 的 COMPUTE_UNIT/SUPPORT_COMPUTE_UNIT 新增 ascend960dt、ascend960pr（TILING_DIR 复用 arch35），op_host/*_def.cpp 新增对应 this->AICore().AddConfig(...) 注册； - 预编译 binary 接入：batch_to_space_nd、concat_d、depth_to_space、pack、space_to_depth、rint 新增 op_host/config/ascend960dt/、op_host/config/ascend960pr/ 下的 *_binary.json 与 *_simplified_key.ini；concat_v2 为 HOSTCPU 算子（依赖 concat），无独立 binary；erfc 的 kernel 源码（arch35/erfc.cpp）通过 add_kernel_sources 的 COMPUTE_UNITS 纳入新平台编译； - concat 的 arch35 tiling/kernel 适配 DAV_9201 架构（仅 9201 生效，不影响既有平台路径）： - tiling 侧：DAV_9201 平台 ubFactorDim0 按两倍 BLOCK（64B）对齐修正；新增 dtypeSize 提升优化（末轴字节满足 128/64/32/16B 对齐时提升 dtypeSize，减少搬运次数），并注册 bitwidth=16 系列 tilingKey（12116/12126/21116/21216/30016 等）； - kernel 侧：__NPU_ARCH__ == 9201 下 DataCopyPad 使用带 enAtomicSync 参数的接口分支。 ## 关联的Issue #3222 https://gitcode.com/cann/ops-math/issues/3222 ## 测试 平台注册与预编译 binary 接入部分不涉及既有 kernel 逻辑修改；concat 为 DAV_9201 架构适配新增专属分支，既有平台路径不变；可通过 PR 评论 compile 触发 CI 编译流水线验证。 ## 文档更新 无 ## 类型标签 - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!5778	29 days ago
pack_def.cpp	[ascend960dt/ascend960pr] conversion/math类9个算子新增ascend960dt/ascend960pr平台支持 Co-authored-by: yangjinwen<yangjinwen2@huawei.com> # message auto-generated for no-merge-commit merge: !5778 merge dev-v121 into master [ascend960dt/ascend960pr] conversion/math类9个算子新增ascend960dt/ascend960pr平台支持 Created-by: yangjinwen Commit-by: yangjinwen Merged-by: cann-robot Description: ## 描述 conversion 类 7 个算子（batch_to_space_nd、concat、concat_d、concat_v2、depth_to_space、pack、space_to_depth）与 math 类 2 个算子（erfc、rint）新增 ascend960dt / ascend960pr 平台支持，复用 arch35 tiling 框架，主要改动： - 平台注册：9 个算子 CMakeLists.txt 的 COMPUTE_UNIT/SUPPORT_COMPUTE_UNIT 新增 ascend960dt、ascend960pr（TILING_DIR 复用 arch35），op_host/*_def.cpp 新增对应 this->AICore().AddConfig(...) 注册； - 预编译 binary 接入：batch_to_space_nd、concat_d、depth_to_space、pack、space_to_depth、rint 新增 op_host/config/ascend960dt/、op_host/config/ascend960pr/ 下的 *_binary.json 与 *_simplified_key.ini；concat_v2 为 HOSTCPU 算子（依赖 concat），无独立 binary；erfc 的 kernel 源码（arch35/erfc.cpp）通过 add_kernel_sources 的 COMPUTE_UNITS 纳入新平台编译； - concat 的 arch35 tiling/kernel 适配 DAV_9201 架构（仅 9201 生效，不影响既有平台路径）： - tiling 侧：DAV_9201 平台 ubFactorDim0 按两倍 BLOCK（64B）对齐修正；新增 dtypeSize 提升优化（末轴字节满足 128/64/32/16B 对齐时提升 dtypeSize，减少搬运次数），并注册 bitwidth=16 系列 tilingKey（12116/12126/21116/21216/30016 等）； - kernel 侧：__NPU_ARCH__ == 9201 下 DataCopyPad 使用带 enAtomicSync 参数的接口分支。 ## 关联的Issue #3222 https://gitcode.com/cann/ops-math/issues/3222 ## 测试 平台注册与预编译 binary 接入部分不涉及既有 kernel 逻辑修改；concat 为 DAV_9201 架构适配新增专属分支，既有平台路径不变；可通过 PR 评论 compile 触发 CI 编译流水线验证。 ## 文档更新 无 ## 类型标签 - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!5778	29 days ago
pack_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	11 days ago