| Files | Last commit | Last update |
|---|---|---|
master翻译合入 Co-authored-by: caiwenwen<caiwenwen6@h-partners.com> # message auto-generated for no-merge-commit merge: !6751 merge master into master master翻译合入 Created-by: caiwenwen Commit-by: caiwenwen Merged-by: cann-robot Description: ## 描述 合入math仓翻译的内容 ## 关联的Issue https://gitcode.com/cann/ops-math/issues/4170 ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 docs/context，aclnn等商发文档 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!6751 | 13 days ago | |
Expand示例改用C++ RAII资源管理，与IsFinite/IsInf示例风格对齐 Co-authored-by: zhouqilong<zhouqilong2@huawei.com> # message auto-generated for no-merge-commit merge: !4682 merge raii-expand into master Expand示例改用C++ RAII资源管理，与IsFinite/IsInf示例风格对齐 Created-by: zhou-qilong Commit-by: zhouqilong Merged-by: cann-robot Description: ## 描述 将 math/expand/examples 下的 aclnn 示例由手动资源管理改为 C++ RAII 管理，与同仓 math/is_finite、math/is_inf 示例风格保持一致。 ## 测试 - bash build.sh --run_example expand eager --soc=ascend950：编译通过，在 Ascend950PR 上运行成功，示例正常退出。 ## 文档更新 无。 ## 类型标签 - [x] 其他，请描述：示例代码改造（C++ RAII资源管理） See merge request: cann/ops-math!4682 | 2 months ago | |
feat: Adds、Expand、BroadcastTo、Tile、GetShape算子新增ascend960dt平台支持 Co-authored-by: tan_xin<tanxin44@huawei.com> # message auto-generated for no-merge-commit merge: !5447 merge master-v121 into master feat: Adds、Expand、BroadcastTo、Tile、GetShape算子新增ascend960dt平台支持 Created-by: tan_xin Commit-by: tan_xin;t00850857 Merged-by: cann-robot Description: ## 描述 为 Adds、Expand、BroadcastTo、Tile、GetShape 五个算子新增 ascend960dt 平台支持，使其可在 ascend960dt（Ascend960DT_968B7）芯片上完成编译与调用。主要改动： ### 算子定义与编译配置 - 各算子 op_host 定义文件（adds_def.cpp、expand_def.cpp、broadcast_to_def.cpp、tile_def.cpp、get_shape_def.cpp）通过 AICore().AddConfig("ascend960dt", ...) 注册平台配置 - 各算子 CMakeLists.txt 的 SUPPORT_COMPUTE_UNIT 增加 "ascend960dt"，tiling 目录沿用 arch35；get_shape 的 kernel 源码 COMPUTE_UNITS 同步增加 ascend960dt - 新增各算子 ascend960dt 平台二进制编译配置 op_host/config/ascend960dt/*_binary.json 与 *_simplified_key.ini（simplified_key_mode=0） - ascendc_config.json 中 Adds、BroadcastTo、Tile 的 compute_units 增加 ascend960dt，并配置对应编译选项 ### 公共构建配置 - 顶层 CMakeLists.txt 的 ASCEND_ALL_COMPUTE_UNIT 增加 ascend960dt - scripts/build.conf.sh 的 SUPPORT_COMPUTE_UNIT_SHORT 增加 ascend960dt - scripts/kernel/binary_script/build_env.sh 的 SOC_MAP 增加 [Ascend960DT]="Ascend960DT_968B7" ### op_api 适配 - expand：aclnn_expand.cpp 的 dataTypeSupportedMap 与 expand.cpp 的 IsAiCoreSupport 增加 NpuArch::DAV_9201 分支，数据类型支持列表沿用 ASCEND910B（FLOAT、FLOAT16、INT32、UINT8、INT8、BOOL、BF16、INT64） - 公共头文件 aclnn_check.h：IsRegBase 的 regbaseArch 集合增加 NpuArch::DAV_9201、NpuArch::DAV_9202；其余为 clang-format 格式整改 ## 关联的Issue - #3561 ## 测试 - 已通过 pre-commit 检查（clang-format 格式、codespell 拼写、文件末尾换行整改，见提交记录） ## 文档更新 - 无 ## 类型标签 - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!5447 | 27 days ago | |
feat(infershape): 补齐 12 个算子 op_graph InferDataType 注册 Co-authored-by: yangjinhan<yangjinhan@huawei.com> # message auto-generated for no-merge-commit merge: !6644 merge feat/register-infer-datatype into master feat(infershape): 补齐 12 个算子 op_graph InferDataType 注册 Created-by: Ruyue1550 Commit-by: yangjinhan Merged-by: cann-robot Description: ## 描述 按 RT1.0 到 RT2.0 推导框架迁移方案第 1 步（补齐缺失的 InferDataType 注册，零行为变更）要求，为 12 个缺失 InferDataType 注册的算子新增 op_graph/*_graph_infer.cpp 注册文件。本次变更只新增文件、不修改任何存量代码，共 12 个文件、240 行。 ### 改动点 | # | 算子 | 新增文件 | 注册方式 | |---|------|---------|---------| | 1 | Pad | conversion/pad/op_graph/pad_graph_infer.cpp | InferDataTypeOutputSameAsInput | | 2 | PadV2 | conversion/pad_v2/op_graph/pad_v2_graph_infer.cpp | 同上 | | 3 | PadV3 | conversion/pad_v3/op_graph/pad_v3_graph_infer.cpp | 同上 | | 4 | PadV3Grad | conversion/pad_v3_grad/op_graph/pad_v3_grad_graph_infer.cpp | 同上 | | 5 | MirrorPad | conversion/mirror_pad/op_graph/mirror_pad_graph_infer.cpp | 同上 | | 6 | BroadcastTo | conversion/broadcast_to/op_graph/broadcast_to_graph_infer.cpp | 同上 | | 7 | Transpose | conversion/transpose/op_graph/transpose_graph_infer.cpp | 同上 | | 8 | Rint | math/rint/op_graph/rint_graph_infer.cpp | 同上 | | 9 | TruncateDiv | math/truncate_div/op_graph/truncate_div_graph_infer.cpp | 同上 | | 10 | Expand | math/expand/op_graph/expand_graph_infer.cpp | 同上 | | 11 | GetShape | math/get_shape/op_graph/get_shape_graph_infer.cpp | 专门推导，输出固定 INT32 | | 12 | HistogramFixedWidth | math/histogram_fixed_width/op_graph/histogram_fixed_width_graph_infer.cpp | 专门推导，输出固定 INT32 | ### 改动原因 1. 背景：RT2.0 V2 路径的 dtype 推导在未注册 InferDataType 时依赖 IR（proto/def 的 TensorType）符号映射兜底。以显式注册替代隐式兜底，使推导逻辑代码自明、不依赖 IR 声明的隐式行为，并为后续推导框架切换就位 op_graph 注册设施。 2. dtype 语义逐算子以 RT1.0 SetDataType 为基线核对（迁移原则：非算子语义变更）： - 上述 #1-#10 共 10 个算子的 canndev RT1.0 实现均为 SetDataType(input_dtype)（取输入 x 的类型，const/非 const/未知 rank 各分支一致），输出 dtype 跟随输入，故注册仓内公共推导函数 InferDataTypeOutputSameAsInput，与本仓已有 Tile/Erfc/TruncateMod 注册同构； - GetShape、HistogramFixedWidth 的 RT1.0 实现硬编码 SetDataType(DT_INT32)，输出为形状值/直方图计数，与输入 dtype 无关，故注册输出固定 INT32 的专门推导，与两者 proto/def 中 OUTPUT(y, TensorType({DT_INT32})) 声明一致。 3. 零行为变更：注册结果与 IR 符号推导兜底结果一致，不改变任何现网推导行为。 4. 构建零改动：op_graph/*.cpp 由 CMake GLOB_RECURSE 自动收集（cmake/func.cmake），无需修改 CMakeLists。 ## 关联的Issue https://gitcode.com/cann/ops-math/issues/4099 ## 测试 - 本地无 CANN 构建环境，未执行编译与 UT，请通过门禁验证（评论 compile 触发 CI） - 已完成静态核对：13 个注册的 dtype 语义与 canndev RT1.0 SetDataType 逐分支一致、与各算子 proto/def 输出声明一致、与本仓已有同款注册（tile/erfc/truncate_mod/circular_pad）代码风格一致 - UT dtype 断言用例（断言注册结果=兜底结果）可随检视意见补充 ## 文档更新 无需更新文档（纯推导注册代码，无接口与行为变化） ## 类型标签 - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!6644 | 13 days ago | |
feat: tile、expand、broadcast_to算子infershape整改 Co-authored-by: ly_cann_coder<liyao152@huawei.com> # message auto-generated for no-merge-commit merge: !6797 merge dev_1030 into master feat: tile、expand、broadcast_to算子infershape整改 Created-by: ly_cann_coder Commit-by: ly_cann_coder Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> tile、expand、broadcast_to算子infershape整改，功能与RT1.0对齐 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> https://gitcode.com/cann/ops-math/issues/4273 ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!6797 | 11 days ago | |
新增Expand算子 Co-authored-by: l00939308<luwenxiang4@huawei.com> # message auto-generated for no-merge-commit merge: !1133 merge master into master 新增Expand算子 Created-by: luwenxiang1998 Commit-by: l00939308 Merged-by: cann-robot Description: ## 描述 新增Expand算子 ## 关联的Issue https://gitcode.com/cann/ops-math/issues/814 ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [x] 其他，请描述：新增Expand算子 See merge request: cann/ops-math!1133 | 8 months ago | |
refactor: 明确 Expand 拷贝策略判断函数命名(#4405) Co-authored-by: ayuemoon<15802951036@163.com> # message auto-generated for no-merge-commit merge: !7002 merge fix/expand-copy-predicate-warning-20261102 into master refactor: 明确 Expand 拷贝策略判断函数命名(#4405) Created-by: ayuemoon Commit-by: ayuemoon Merged-by: cann-robot Description: ## 描述 Expand 的 ShouldUseBiggerMemCpy 被 G.FUU.13「禁止封装安全函数」报告为疑似自定义风险函数。该函数实际上仅比较字节数阈值，不执行拷贝。 将其改名为 ShouldUseBulkCopy，明确表达拷贝策略选择的含义，同步主模板、half/bfloat16 两处特化和调用点，共 4 处。函数体和计算逻辑保持原样。 本次未做平台屏蔽，告警是否消除需 CodeCheck 复扫确认。 ## 关联的Issue https://gitcode.com/cann/ops-math/issues/4405 ## 测试 - 编译验证：成功构建 math_aicpu_op_kernel_ut 目标。 - 单元测试：执行 TEST_EXPAND_UT.*，共 9 个用例，全部通过。 - 覆盖 Expand 的相关功能场景，包括使用输入维度处理 -1、拒绝尺寸不足的输入等。 ## 文档更新 无。 ## 类型标签 - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [x] 其他，请描述：内部函数命名整理，处理静态检查反馈。 See merge request: cann/ops-math!7002 | 8 days ago | |
feat: tile、expand、broadcast_to算子infershape整改 Co-authored-by: ly_cann_coder<liyao152@huawei.com> # message auto-generated for no-merge-commit merge: !6797 merge dev_1030 into master feat: tile、expand、broadcast_to算子infershape整改 Created-by: ly_cann_coder Commit-by: ly_cann_coder Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> tile、expand、broadcast_to算子infershape整改，功能与RT1.0对齐 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> https://gitcode.com/cann/ops-math/issues/4273 ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!6797 | 11 days ago | |
feat: expand/tile/pad/pad_v3/mirror_pad/get_shape支持Ascend960PR Co-authored-by: ly_cann_coder<liyao152@huawei.com> # message auto-generated for no-merge-commit merge: !6568 merge dev_960 into master feat: expand/tile/pad/pad_v3/mirror_pad/get_shape支持Ascend960PR Created-by: ly_cann_coder Commit-by: ly_cann_coder Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> expand/tile/pad/pad_v3/mirror_pad/get_shape支持Ascend960PR ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> https://gitcode.com/cann/ops-math/issues/3836 ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!6568 | 18 days ago | |
产品名变更 Co-authored-by: gitcode-chenjiao<chenjiao31@huawei.com> # message auto-generated for no-merge-commit merge: !5912 merge master into master 产品名变更 Created-by: gitcode-chenjiao Commit-by: gitcode-chenjiao Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> AI处理器对应的产品型号名称更新 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> [#3258](https://gitcode.com/cann/ops-math/issues/3258) ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ok ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> *.md ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!5912 | 1 month ago |
Expand
产品支持情况
| 产品 | 是否支持 |
|---|---|
| Ascend 950PR&950DT系列产品 | √ |
| Atlas A3系列产品 | √ |
| Atlas A2系列产品 | √ |
| Atlas 200I/500 A2推理产品 | × |
| Atlas推理系列产品 | √ |
| Atlas训练系列产品 | √ |
功能说明
- 算子功能：将输入tensor广播到指定的shape。如输入tensor的shape为(1, 4)，指定的shape为(2, 4)，则输出是shape为(2, 4)的tensor。
参数说明
| 参数名 | 输入/输出/属性 | 描述 | 数据类型 | 数据格式 |
|---|---|---|---|---|
| x | 输入张量 | 需要被广播的张量。 | FLOAT16、FLOAT、INT32、INT64、INT8、UINT8、BOOL、BF16 | ND |
| shape | 输入张量 | 表示x广播后的shape大小。 | INT64、INT32 | - |
| y | 输出 | 维度最大不超过8维，shape由shape输入决定，dtype需要与self一致。 | 同x | ND |
约束说明
无
调用说明
| 调用方式 | 样例代码 | 说明 |
|---|---|---|
| 图模式调用 | test_geir_expand | 通过GE IR图模式调用Expand算子 |
| aclnn接口 | test_aclnn_expand | 通过aclnnExpand接口方式调用Expand算子。 |