Star0

Star0

cann-robotKirin X90/9030支持Cast/Mul/Transpose/Add/ConcatD/StridedSlice算子

32c65263Add on 12 days agoHistory

Files	Last commit	Last update
docs	math仓英文合入 Co-authored-by: y60124828<yanglou1@h-partners.com> # message auto-generated for no-merge-commit merge: !6772 merge master into master math仓英文合入 Created-by: yanglu-1 Commit-by: y60124828 Merged-by: cann-robot Description: ## 描述 math仓master分支英文合入 ## 关联的Issue 关联issue：[4170](https://gitcode.com/cann/ops-math/issues/4170) ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!6772	12 days ago
examples	【bugfix】修改文件夹名称、注释拼写问题 Co-authored-by: majiajian_hw<majiajian@huawei.com> # message auto-generated for no-merge-commit merge: !4112 merge bugfix_md0716 into master 【bugfix】修改文件夹名称、注释拼写问题 Created-by: nunnons2 Commit-by: majiajian_hw Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> 修改文件夹名称、注释拼写问题 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> https://gitcode.com/cann/ops-math/issues/1705 ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> 已完成 ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> NA ## 类型标签 <!-- [x] 表示选中 --> - [x] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!4112	3 months ago
framework	refactor: npu_indexing/correlation/one_hot_d onnx 插件迁移至 ParseParamsByOperatorFn 接口 Co-authored-by: clinglai0517<laichangling@huawei.com> # message auto-generated for no-merge-commit merge: !5325 merge fix-onnx-partitionedcall into master refactor: npu_indexing/correlation/one_hot_d onnx 插件迁移至 ParseParamsByOperatorFn 接口 Created-by: clinglai0517 Commit-by: clinglai0517 Merged-by: cann-robot Description: ## 描述将 npu_indexing、correlation、one_hot_d 三个 ONNX 插件从旧 ParseParamsFn(const Message, ge::Operator&) 接口迁移至 ParseParamsByOperatorFn，去除onnx_common.h / ge::onnx:: / NodeProto 依赖（强依赖 ge_onnx.pb.h），插件与 framework UT 编译均脱离 protobuf。 ### 迁移内容 - conversion/strided_slice/framework/npu_indexing_onnx_plugin.cpp + 7 UT ：begins/ends/strides 与五个 mask 经 json 解析。 - common/src/framework/correlation_onnx_plugin.cpp + 5 UT ：groups 属性经 json 解析，保留默认值。 - math/one_hot/framework/one_hot_d_onnx_plugin.cpp + 7 UT ：depth（必需）/ num_classes / on_value / off_value 经 json 解析。 ### 公共工具下沉 onnx_common.h 中所有无 protobuf 依赖的 helper 全部下沉至common/inc/framework/math_onnx_plugin_util.h（protobuf-free）：Vec2Tensor、CreateScalar、DataTypeOnnx、onnx2om_dtype_map、GetOmDtypeFromOnnxDtype、ChangeFormatFromOnnx。onnx_common.h 现仅保留ge_onnx.pb.h 等 include 链供未迁移插件使用，经包含 util 仍提供这些 helper，无破坏。 ### CI 修复 1. scripts/ci/gen_ci_cmd.py：get_op_ut_types 识别 <op>/framework/ 与 <op>/tests/ut/framework/ 变更为 ophost，修复仅改 framework/onnx 插件的 PR 生成 0 条命令、UT 未运行的问题。 2. cmake/ut.cmake：framework 分支 --ops 过滤区分 common 集中目录（按文件名）与独立算子目录（按目录算子名），修复 test__onnx_plugin.cpp 因文件名算子名与目录算子名不一致被 continue 跳过、framework_ut_cases 仅剩 empty.cpp 占位的问题。 ## 关联的Issue Issue #2892 ## 测试 - 容器 CANN 9.0.0 / Ascend 910B3：Onnx*PluginTest 共 19 用例全 [ PASSED ]。 - CI 增量 --ops=correlation,one_hot,strided_slice --ophost：46 用例（含 19 个 onnx framework 用例）全过。 - pre-commit（clang-format / end-of-file / OAT）全绿。 ## 文档更新不涉及 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!5325	2 months ago
op_api	slice strided_slice strided_slice_v3注册补全HIFLOAT4/HIFLOAT4_SCALE类型 Co-authored-by: xiapengcheng5<xiapengcheng5@huawei.com> # message auto-generated for no-merge-commit merge: !6471 merge master into master slice strided_slice strided_slice_v3注册补全HIFLOAT4/HIFLOAT4_SCALE类型 Created-by: xiapengcheng5 Commit-by: xiapengcheng5 Merged-by: cann-robot Description: ## 描述 slice strided_slice strided_slice_v3注册补全HIFLOAT4/HIFLOAT4_SCALE类型 ## 关联的Issue [#3659](https://gitcode.com/cann/ops-math/issues/3659) ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新无 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!6471	21 days ago
op_graph	slice strided_slice strided_slice_v3注册补全HIFLOAT4/HIFLOAT4_SCALE类型 Co-authored-by: xiapengcheng5<xiapengcheng5@huawei.com> # message auto-generated for no-merge-commit merge: !6471 merge master into master slice strided_slice strided_slice_v3注册补全HIFLOAT4/HIFLOAT4_SCALE类型 Created-by: xiapengcheng5 Commit-by: xiapengcheng5 Merged-by: cann-robot Description: ## 描述 slice strided_slice strided_slice_v3注册补全HIFLOAT4/HIFLOAT4_SCALE类型 ## 关联的Issue [#3659](https://gitcode.com/cann/ops-math/issues/3659) ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新无 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!6471	21 days ago
op_host	Kirin X90/9030支持Cast/Mul/Transpose/Add/ConcatD/StridedSlice算子 Co-authored-by: chenlong1<chenlong1@huawei.com> Co-authored-by: jinzy1999<jinzy@pku.edu.cn> # message auto-generated for no-merge-commit merge: !6581 merge master into master Kirin X90/9030支持Cast/Mul/Transpose/Add/ConcatD/StridedSlice算子 Created-by: chenlong1 Commit-by: chenlong1;jinzy1999 Merged-by: cann-robot Description: ## 描述 Kirin X90/9030支持Cast/Mul/Transpose/Add/ConcatD/StridedSlice算子 ## 关联的Issue https://gitcode.com/cann/ops-math/issues/3790 ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!6581	12 days ago
op_kernel	Kirin X90/9030支持Cast/Mul/Transpose/Add/ConcatD/StridedSlice算子 Co-authored-by: chenlong1<chenlong1@huawei.com> Co-authored-by: jinzy1999<jinzy@pku.edu.cn> # message auto-generated for no-merge-commit merge: !6581 merge master into master Kirin X90/9030支持Cast/Mul/Transpose/Add/ConcatD/StridedSlice算子 Created-by: chenlong1 Commit-by: chenlong1;jinzy1999 Merged-by: cann-robot Description: ## 描述 Kirin X90/9030支持Cast/Mul/Transpose/Add/ConcatD/StridedSlice算子 ## 关联的Issue https://gitcode.com/cann/ops-math/issues/3790 ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!6581	12 days ago
op_kernel_aicpu	fix(aicpu): 清理 conversion 算子 CodeCheck 告警 Co-authored-by: Ding_Jing<dingjing19@huawei.com> # message auto-generated for no-merge-commit merge: !5836 merge fix/codecheck-rules into master fix(aicpu): 清理 conversion 算子 CodeCheck 告警 Created-by: Ding_Jing Commit-by: Ding_Jing Merged-by: cann-robot Description: ## 描述清理 conversion 目录中已确认成立的 AICPU CodeCheck 告警，并保持修改范围仅限告警行及其必要声明： - 将 5 处 `void *` 转换改为仓内统一的 `aicpu::PtrToPtr<void, T>`。 - 为不修改成员状态的报告函数补充 `const`，声明与定义同步。 - 将位掩码载体改为无符号类型，并显式布尔化位运算条件。 - 将报告中的默认 lambda 捕获改为显式捕获，虚析构补充 `override`。 - 两个存量类型分派宏保持与 master 一致；对应 3 条 G.PRE.10 告警确认成立，按最小 diff 原则改为文件/报告行级精确屏蔽，不调整宏定义及 32 个调用点。未扩展清理同文件中未上报的存量写法，未修改 ace 仓代码。 CI 格式门禁修复（2026-09-19）：三次流水线均因 `CodeCheck_precommit` 失败。钩子对已改动文件执行全文件 clang-format 18.1.8，命中 `strided_slice_aicpu.h` 等文件的存量格式。只在本 PR 原有 9 个文件中，对 4 个实际不合规文件做了格式工具生成的机械整理；其中头文件产生约 620 行格式 diff，未额外清理告警或修改计算语义。 ## 关联的Issue 无。本次为 CodeCheck 告警清理，按约定不创建 Issue。 ## 测试 - 基于最新 `origin/master` 执行 AICPU UT，5 个测试套 105/105 通过。 - 最终改为 `PtrToPtr` 后重新编译并执行同一目标，仍为 105/105 通过。 - 构建参数包含 `-Werror -ftrapv`。 - `git diff --check origin/master...HEAD` 通过。 - 当前提交的服务端 CodeCheck/CI 待完成 3 条 G.PRE.10 精确屏蔽后确认。 - 基于最新 master 重新执行 9 个改动文件的完整 pre-commit（含格式、尾随空白与 OAT），全部通过。 - 宏修改撤回后重新执行完整 pre-commit，全部通过；重新编译并运行 5 个目标 AICPU 测试套，105/105 通过。 - 上一版服务端 run #3847 为 52/52 job 通过；本版 3 条 G.PRE.10 待按精确屏蔽理由处理后重跑 CI。 ## 文档更新无。本次不涉及用户接口或功能语义变更。 ## 类型标签 - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [x] 其他，请描述：CodeCheck 告警清理 See merge request: cann/ops-math!5836	1 month ago
tests	refactor: npu_indexing/correlation/one_hot_d onnx 插件迁移至 ParseParamsByOperatorFn 接口 Co-authored-by: clinglai0517<laichangling@huawei.com> # message auto-generated for no-merge-commit merge: !5325 merge fix-onnx-partitionedcall into master refactor: npu_indexing/correlation/one_hot_d onnx 插件迁移至 ParseParamsByOperatorFn 接口 Created-by: clinglai0517 Commit-by: clinglai0517 Merged-by: cann-robot Description: ## 描述将 npu_indexing、correlation、one_hot_d 三个 ONNX 插件从旧 ParseParamsFn(const Message, ge::Operator&) 接口迁移至 ParseParamsByOperatorFn，去除onnx_common.h / ge::onnx:: / NodeProto 依赖（强依赖 ge_onnx.pb.h），插件与 framework UT 编译均脱离 protobuf。 ### 迁移内容 - conversion/strided_slice/framework/npu_indexing_onnx_plugin.cpp + 7 UT ：begins/ends/strides 与五个 mask 经 json 解析。 - common/src/framework/correlation_onnx_plugin.cpp + 5 UT ：groups 属性经 json 解析，保留默认值。 - math/one_hot/framework/one_hot_d_onnx_plugin.cpp + 7 UT ：depth（必需）/ num_classes / on_value / off_value 经 json 解析。 ### 公共工具下沉 onnx_common.h 中所有无 protobuf 依赖的 helper 全部下沉至common/inc/framework/math_onnx_plugin_util.h（protobuf-free）：Vec2Tensor、CreateScalar、DataTypeOnnx、onnx2om_dtype_map、GetOmDtypeFromOnnxDtype、ChangeFormatFromOnnx。onnx_common.h 现仅保留ge_onnx.pb.h 等 include 链供未迁移插件使用，经包含 util 仍提供这些 helper，无破坏。 ### CI 修复 1. scripts/ci/gen_ci_cmd.py：get_op_ut_types 识别 <op>/framework/ 与 <op>/tests/ut/framework/ 变更为 ophost，修复仅改 framework/onnx 插件的 PR 生成 0 条命令、UT 未运行的问题。 2. cmake/ut.cmake：framework 分支 --ops 过滤区分 common 集中目录（按文件名）与独立算子目录（按目录算子名），修复 test__onnx_plugin.cpp 因文件名算子名与目录算子名不一致被 continue 跳过、framework_ut_cases 仅剩 empty.cpp 占位的问题。 ## 关联的Issue Issue #2892 ## 测试 - 容器 CANN 9.0.0 / Ascend 910B3：Onnx*PluginTest 共 19 用例全 [ PASSED ]。 - CI 增量 --ops=correlation,one_hot,strided_slice --ophost：46 用例（含 19 个 onnx framework 用例）全过。 - pre-commit（clang-format / end-of-file / OAT）全绿。 ## 文档更新不涉及 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!5325	2 months ago
CMakeLists.txt	Kirin X90/9030支持Cast/Mul/Transpose/Add/ConcatD/StridedSlice算子 Co-authored-by: chenlong1<chenlong1@huawei.com> Co-authored-by: jinzy1999<jinzy@pku.edu.cn> # message auto-generated for no-merge-commit merge: !6581 merge master into master Kirin X90/9030支持Cast/Mul/Transpose/Add/ConcatD/StridedSlice算子 Created-by: chenlong1 Commit-by: chenlong1;jinzy1999 Merged-by: cann-robot Description: ## 描述 Kirin X90/9030支持Cast/Mul/Transpose/Add/ConcatD/StridedSlice算子 ## 关联的Issue https://gitcode.com/cann/ops-math/issues/3790 ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!6581	12 days ago
README.md	Kirin X90/9030支持Cast/Mul/Transpose/Add/ConcatD/StridedSlice算子 Co-authored-by: chenlong1<chenlong1@huawei.com> Co-authored-by: jinzy1999<jinzy@pku.edu.cn> # message auto-generated for no-merge-commit merge: !6581 merge master into master Kirin X90/9030支持Cast/Mul/Transpose/Add/ConcatD/StridedSlice算子 Created-by: chenlong1 Commit-by: chenlong1;jinzy1999 Merged-by: cann-robot Description: ## 描述 Kirin X90/9030支持Cast/Mul/Transpose/Add/ConcatD/StridedSlice算子 ## 关联的Issue https://gitcode.com/cann/ops-math/issues/3790 ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!6581	12 days ago

StridedSlice

产品支持情况

产品	是否支持
Ascend 950PR&950DT系列产品	√
Atlas A3系列产品	√
Atlas A2系列产品	√
Atlas 200I/500 A2推理产品	√
Atlas推理系列产品	√
Atlas训练系列产品	√
Kirin X90处理器系列产品	√
Kirin 9030处理器系列产品	√

功能说明

算子功能：按照指定的起始、结束位置和步长，从输入张量中提取一个子张量。

参数说明

参数名	输入/输出/属性	描述	数据类型	数据格式
x	输入	输入的张量	INT8、UINT8、INT16、UINT16、INT32、UINT32、INT64、UINT64、FLOAT、FLOAT16、BF16、BOOL、COMPLEX32、COMPLEX64、HIFLOAT8、FLOAT8_E5M2、FLOAT8_E4M3FN	ND
begin	输入	每个维度的起始值	INT8、UINT8、INT16、UINT16、INT32、UINT32、INT64、UINT64、FLOAT、FLOAT16、BF16、BOOL、COMPLEX32、COMPLEX64、HIFLOAT8、FLOAT8_E5M2、FLOAT8_E4M3FN	ND
end	输入	每个维度的结束值	INT8、UINT8、INT16、UINT16、INT32、UINT32、INT64、UINT64、FLOAT、FLOAT16、BF16、BOOL、COMPLEX32、COMPLEX64、HIFLOAT8、FLOAT8_E5M2、FLOAT8_E4M3FN	ND
strides	输入	每个维度上每个点取值的跨度	INT8、UINT8、INT16、UINT16、INT32、UINT32、INT64、UINT64、FLOAT、FLOAT16、BF16、BOOL、COMPLEX32、COMPLEX64、HIFLOAT8、FLOAT8_E5M2、FLOAT8_E4M3FN	ND
begin_mask	属性	这个值指定哪些维度的begin被忽略	INT32、INT64	ND
end_mask	属性	这个值指定哪些维度的end被忽略	INT32、INT64	ND
ellipsis_mask	属性	从指定的维度开始全选，直到遇到用户指定begin才退出	INT32、INT64	ND
new_axis_mask	属性	在指定位上增加维度为1的shape	INT32、INT64	ND
shrink_axis_mask	属性	把对应索引处维度强制降为1	INT32、INT64	ND
y	输出	输出张量	INT8、UINT8、INT16、UINT16、INT32、UINT32、INT64、UINT64、FLOAT、FLOAT16、BF16、BOOL、COMPLEX32、COMPLEX64、HIFLOAT8、FLOAT8_E5M2、FLOAT8_E4M3FN	ND

约束说明

Kirin X90/Kirin 9030系列 x/y 仅支持INT8、UINT8、INT16、UINT16、INT32、FLOAT16、FLOAT32数据类型；begin/end/strides 仅支持INT32数据类型

调用说明

调用方式	调用样例	说明
aclnn调用	test_aclnn_strided_slice	通过aclnnStridedSlice接口方式调用StridedSlice算子。
图模式调用	test_strided_slice	通过算子IR构图方式调用StridedSlice算子