| 文件 | 最后提交记录 | 最后更新时间 |
|---|---|---|
refactor: 迁移 arch35 算子 Reg API、修正拼写并规范化日志文案 Co-authored-by: esok11<yangsifa@huawei.com> # message auto-generated for no-merge-commit merge: !5135 merge f082601 into master refactor: 迁移 arch35 算子 Reg API、修正拼写并规范化日志文案 Created-by: esok11 Commit-by: esok11 Merged-by: cann-robot Description: ## 描述 对 arch35 架构下多个算子进行 AscendC Reg API 迁移、命名拼写修正及日志文案规范化，共 46 个文件，涉及 conversion/ 与 math/ 下 15 个算子。 主要改动： - AscendC Reg API 迁移（arch35 kernel 代码，21 个 .h 文件）： - 地址空间修饰符： __local_mem__ → __ubuf__ - Reg 数据搬运 API 细化：加载用 AscendC::Reg::LoadAlign（原 Reg::DataCopy 加载语义），存储用 AscendC::Reg::StoreAlign（原 Reg::DataCopy 存储语义） - 掩码操作 API 重命名：AscendC::Reg::MaskNot → AscendC::Reg::Not、AscendC::Reg::MaskAnd → AscendC::Reg::And - 移除冗余 using AscendC::Reg::DataCopy; 声明 - 命名拼写修正（op_host/arch35，3 个文件）： - 方法名 TilingForReleatedTranspose → TilingForRelatedTranspose（Releated→Related） - 成员变量 isReleatedTranspsoe_ → isRelatedTranspose_（Releated→Related, Transpsoe→Transpose） - 同步修改 confusion_transpose_d、depth_to_space 中对该方法的调用 - 日志清理（examples、op_host、op_api，22 个文件）： - 失败路径日志级别 INFO → ERROR（Initialize ge failed、Run graph failed、Finalize ir graph session failed 等） - 文案语法修正：only support → only supports、is: %s and %s → are: %s and %s、expect %f → expected %f、multiples's → multiples' - tile 示例代码宏定义缩进格式化 ### 改动原因 配合 AscendC Reg API 演进，将 arch35 算子迁移至新的地址空间修饰符与 LoadAlign/StoreAlign/Not/And API；修正 transpose 相关命名中的拼写错误；统一日志级别与文案规范，提升日志可读性与问题定位效率。 ### 改动方法 - kernel 侧：逐文件替换 __local_mem__→__ubuf__、Reg::DataCopy（加载）→Reg::LoadAlign、Reg::DataCopy（存储）→Reg::StoreAlign、Reg::MaskNot→Reg::Not、Reg::MaskAnd→Reg::And，不改变控制流。 - host 侧：重命名 TilingForReleatedTranspose→TilingForRelatedTranspose、isReleatedTranspsoe_→isRelatedTranspose_，同步所有调用点。 - 日志：修改 printf/OP_LOGE/OP_LOGW/OP_LOGI 文案与级别字符串，不改变校验逻辑。 ## 关联的Issue - #2916 ## 测试 本 PR 为 API 迁移、命名修正与日志清理，通过已有测试用例（含 arch35 tiling UT 与 examples 二级冒烟）回归验证。 ## 文档更新 本 PR 未包含文档更新。 ## 类型标签 - [ ] Bug修复 - [ ] 新特性 - [x] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述：AscendC Reg API 迁移 + 命名拼写修正 + 日志规范化 See merge request: cann/ops-math!5135 | 1 个月前 | |
FIx the null-check macros for the five operators Co-authored-by: yangjinhan<yangjinhan@huawei.com> # message auto-generated for no-merge-commit merge: !4561 merge fix-add-null-check-macro into master FIx the null-check macros for the five operators Created-by: Ruyue1550 Commit-by: yangjinhan Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> 补全5个算子(MatrixSetDiagV2、Erfc、GetShape、HistogramFixedWidth、TileWithAxis)的判空宏 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [x] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!4561 | 2 个月前 | |
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 | 2 天前 | |
getshape算子补充host侧数据校验，以及readme修改 Co-authored-by: t00850857<tanxin44@huawei.com> # message auto-generated for no-merge-commit merge: !4010 merge master-getshape-2 into master getshape算子补充host侧数据校验，以及readme修改 Created-by: tan_xin Commit-by: t00850857 Merged-by: cann-robot Description: ## 描述 getshape算子补充host侧rank未知，输出长度超过128，空tensorlist等数据校验，以及readme修改 ## 关联的Issue https://gitcode.com/cann/ops-math/issues/2408 ## 测试 二级冒烟，泛化用例验证 ## 文档更新 更新了getshape算子readme.md ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!4010 | 2 个月前 | |
test: 18 算子 golden 新增 TestSpec 双注册与 third_party Co-authored-by: Xbying<2119046648@qq.com> # message auto-generated for no-merge-commit merge: !5789 merge 14 into master test: 18 算子 golden 新增 TestSpec 双注册与 third_party Created-by: 2301_81474000 Commit-by: Xbying Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> - 16 个算子 golden 新增 __spec__（保留旧 __golden__，单一代码源双入口）：Pad、PadV2、PadV3、MirrorPad、CircularPadGrad、PadV3GradReplication、DepthToSpace、SpaceToDepth、BatchToSpaceND、MatrixSetDiagV2、Sinc、TruncateDiv、TruncateMod、Erfc、Expand、Tile/TileD - 新建 2 个 golden：ReflectionPad3dGrad、Rint - 全部补 third_party 竞品实现（F.pad / aten backward / np.pad / torch.tile / torch.round 等），视图类输出加 .contiguous() - 顺带修复 4 处原 golden 缺陷（depth_to_space 死链与未定义名、pad_v2 0 维标量越界、pad_v3_grad_replication 部分维 pad、ruff E712/F841） - 全部经 TTK loader 双入口解析与 third_party↔golden 数值互验（BatchToSpaceND 因 golden 依赖 TF 仅做解析验证） ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> https://gitcode.com/cann/ops-math/issues/3226 ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [x] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!5789 | 4 天前 | |
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 | 2 天前 | |
产品名变更 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 | 24 天前 |
GetShape
产品支持情况
| 产品 | 是否支持 |
|---|---|
| Ascend 950PR&950DT系列产品 | √ |
| Atlas A3系列产品 | √ |
| Atlas A2系列产品 | √ |
| Atlas 200I/500 A2推理产品 | × |
| Atlas推理系列产品 | √ |
| Atlas训练系列产品 | √ |
功能说明
-
算子功能：获取一个或多个输入 tensor 的 shape 信息，按输入顺序将各维度的尺寸值拼接为一维 int32 tensor 输出。
-
计算逻辑：
- 输入 N 个 tensor，第 i 个 tensor 的 rank 为 r_i
- 输出为一维 int32 tensor，长度为 sum(r_i)，内容为所有输入 tensor 各维度值的顺序拼接
-
示例：
输入: x0 = float32[2, 3, 4], x1 = float32[5, 6] 输出: y = int32[5] = [2, 3, 4, 5, 6]
参数说明
| 参数名 | 输入/输出/属性 | 描述 | 数据类型 | 数据格式 |
|---|---|---|---|---|
| x | 动态输入 | 待获取 shape 信息的输入 tensor 列表。 | DOUBLE、FLOAT、FLOAT16、INT8、UINT8、INT16、UINT16、INT32、UINT32、INT64、UINT64、BOOL | ND |
| y | 输出 | 所有输入 tensor 各维度值的顺序拼接，长度为各输入 rank 之和。 | INT32 | ND |
约束说明
- 每个输入 tensor 的 rank 不超过 8。
- 所有输入 tensor 的 rank 之和（即输出长度）不超过 128，且大于 0（不允许所有输入均为 scalar）。
- 输出 dtype 固定为 INT32，不可配置。
- 输入仅支持 ND 格式。
调用说明
| 调用方式 | 调用样例 | 说明 |
|---|---|---|
| 图模式调用 | test_geir_get_shape | 通过算子IR构图方式调用GetShape算子。 |