0

0

cann-robotrectify examples

b0070c68创建于 5 天前历史提交

文件	最后提交记录	最后更新时间
examples	rectify examples Co-authored-by: sujunwei3<sujunwei3@huawei.com> # message auto-generated for no-merge-commit merge: !2972 merge dev_extra into master rectify examples Created-by: sujunwei3 Commit-by: sujunwei3 Merged-by: cann-robot Description: ## 描述整改 GE IR 用例代码，修复部分数据类型下GenData逻辑问题。主要改动： 1. 修复命名空间问题：`FORMAT_ND` → `ge::FORMAT_ND` 2. 修复宏参数类型：`GenOnesData(..., 1)` → `GenOnesData(..., 1.0)` 3. 使用 `std::to_string` 替代字符串拼接，防止未定义行为 4. 删除冗余宏定义（ADD_INPUT_DOUBLE 等），提高代码可维护性 5. 添加 `#include <ctime>` 头文件涉及算子： - slice_write - split_d - unpack - cosh - tile ## 关联的Issue #1694 ## 测试根据代码变更，测试场景如下： 1. GE IR 编译测试 - 验证整改后的 GE IR 用例能够正确编译 - 检查命名空间修复是否生效 2. 算子功能验证 - slice_write: 执行 GE IR 用例，验证切片写入功能 - split_d: 执行 GE IR 用例，验证张量分割功能 - unpack: 执行 GE IR 用例，验证解包功能 - cosh: 执行 GE IR 用例，验证双曲余弦功能 - tile: 执行 GE IR 用例，验证张量复制功能 3. 二级冒烟测试 - 验证算子包编译成功 - 验证 GE IR 执行成功，输出正确 ## 文档更新无文档更新，仅整改 GE IR 用例代码 ## 类型标签 - [ ] Bug修复 - [x] 其他，请描述：GE IR 用例整改 See merge request: cann/ops-math!2972	5 天前
op_graph	增加split_v算子aicpu实现 Co-authored-by: 苏俊伟<sujunwei3@huawei.com> # message auto-generated for no-merge-commit merge: !2275 merge dev into master 增加split_v算子aicpu实现 Created-by: sujunwei3 Commit-by: 苏俊伟 Merged-by: cann-robot Description: ## 描述新增split_v、split_d算子的AI CPU kernel实现，扩展CANN算子库在AI CPU上的算子支持范围。 ## 关联的Issue #1444 ## 测试根据代码变更，测试场景如下： 1. 算子构建测试 - 执行 cmake 配置和编译，验证 CMake 配置变更正确 2. 算子加载测试 - 验证 split_d.json 和 split_v.json 接口注册成功 3. 单元测试 - split_v UT: `bash build.sh -u --opkernel_aicpu --ops=split_v --soc=ascend910b` - split_d UT: `bash build.sh -u --opkernel_aicpu --ops=split_d --soc=ascend910b` 4. GE IR验证 - 编包 → 装包 → 执行 GE IR用例流程通过 ## 文档更新 - 新增 conversion/split_d/README.md，说明split_d算子的设计文档 - docs/zh/op_list.md新增split_v, split_d aicpu算子支持说明 ## 类型标签 - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!2275	1 个月前
op_kernel_aicpu	增加slice_write/lin_space/cast算子aicpu实现 Co-authored-by: sujunwei3<sujunwei3@huawei.com> # message auto-generated for no-merge-commit merge: !2677 merge dev into master 增加slice_write/lin_space/cast算子aicpu实现 Created-by: sujunwei3 Commit-by: sujunwei3 Merged-by: cann-robot Description: ## 描述新增 slice_write、lin_space、cast 三个算子的 AICPU 实现。主要变更： - slice_write 算子: 新增 AICPU 实现，支持多维切片写入操作 - lin_space 算子: 新增 AICPU 实现，生成线性序列 - cast 算子: 新增 AICPU 实现，支持多种数据类型转换 - split_d/split_v 算子: 移除 json 配置文件，统一使用 _aicpu_def.cpp 定义算子原型 ## 关联的Issue #1554 ## 测试根据代码变更，测试场景如下： 1. slice_write 算子测试* - 执行 slice_write AICPU 算子单元测试 - 验证多维切片写入功能正确性 - 边界条件测试 2. lin_space 算子测试 - 执行 lin_space AICPU 算子单元测试 - 验证线性序列生成功能正确性 - 不同数据类型测试 3. cast 算子测试 - 执行 cast AICPU 算子单元测试 - 验证数据类型转换功能正确性 - 各数据类型组合测试 4. split_d/split_v 算子测试 - 验证移除 json 配置后算子仍正常工作 - 执行原有单元测试验证功能不变 5. 编译构建测试 - 执行 cmake 配置和编译，验证新增算子正确构建 - 验证 aicpu_kernels.so 正常链接 ## 文档更新 - 新增 conversion/slice_write/README.md，描述 slice_write 算子使用方法 - 新增各算子单元测试文件 ## 类型标签 - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!2677	18 天前
tests	增加slice_write/lin_space/cast算子aicpu实现 Co-authored-by: sujunwei3<sujunwei3@huawei.com> # message auto-generated for no-merge-commit merge: !2677 merge dev into master 增加slice_write/lin_space/cast算子aicpu实现 Created-by: sujunwei3 Commit-by: sujunwei3 Merged-by: cann-robot Description: ## 描述新增 slice_write、lin_space、cast 三个算子的 AICPU 实现。主要变更： - slice_write 算子: 新增 AICPU 实现，支持多维切片写入操作 - lin_space 算子: 新增 AICPU 实现，生成线性序列 - cast 算子: 新增 AICPU 实现，支持多种数据类型转换 - split_d/split_v 算子: 移除 json 配置文件，统一使用 _aicpu_def.cpp 定义算子原型 ## 关联的Issue #1554 ## 测试根据代码变更，测试场景如下： 1. slice_write 算子测试* - 执行 slice_write AICPU 算子单元测试 - 验证多维切片写入功能正确性 - 边界条件测试 2. lin_space 算子测试 - 执行 lin_space AICPU 算子单元测试 - 验证线性序列生成功能正确性 - 不同数据类型测试 3. cast 算子测试 - 执行 cast AICPU 算子单元测试 - 验证数据类型转换功能正确性 - 各数据类型组合测试 4. split_d/split_v 算子测试 - 验证移除 json 配置后算子仍正常工作 - 执行原有单元测试验证功能不变 5. 编译构建测试 - 执行 cmake 配置和编译，验证新增算子正确构建 - 验证 aicpu_kernels.so 正常链接 ## 文档更新 - 新增 conversion/slice_write/README.md，描述 slice_write 算子使用方法 - 新增各算子单元测试文件 ## 类型标签 - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!2677	18 天前
CMakeLists.txt	增加split_v算子aicpu实现 Co-authored-by: 苏俊伟<sujunwei3@huawei.com> # message auto-generated for no-merge-commit merge: !2275 merge dev into master 增加split_v算子aicpu实现 Created-by: sujunwei3 Commit-by: 苏俊伟 Merged-by: cann-robot Description: ## 描述新增split_v、split_d算子的AI CPU kernel实现，扩展CANN算子库在AI CPU上的算子支持范围。 ## 关联的Issue #1444 ## 测试根据代码变更，测试场景如下： 1. 算子构建测试 - 执行 cmake 配置和编译，验证 CMake 配置变更正确 2. 算子加载测试 - 验证 split_d.json 和 split_v.json 接口注册成功 3. 单元测试 - split_v UT: `bash build.sh -u --opkernel_aicpu --ops=split_v --soc=ascend910b` - split_d UT: `bash build.sh -u --opkernel_aicpu --ops=split_d --soc=ascend910b` 4. GE IR验证 - 编包 → 装包 → 执行 GE IR用例流程通过 ## 文档更新 - 新增 conversion/split_d/README.md，说明split_d算子的设计文档 - docs/zh/op_list.md新增split_v, split_d aicpu算子支持说明 ## 类型标签 - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!2275	1 个月前
README.md	增加split_v算子aicpu实现 Co-authored-by: 苏俊伟<sujunwei3@huawei.com> # message auto-generated for no-merge-commit merge: !2275 merge dev into master 增加split_v算子aicpu实现 Created-by: sujunwei3 Commit-by: 苏俊伟 Merged-by: cann-robot Description: ## 描述新增split_v、split_d算子的AI CPU kernel实现，扩展CANN算子库在AI CPU上的算子支持范围。 ## 关联的Issue #1444 ## 测试根据代码变更，测试场景如下： 1. 算子构建测试 - 执行 cmake 配置和编译，验证 CMake 配置变更正确 2. 算子加载测试 - 验证 split_d.json 和 split_v.json 接口注册成功 3. 单元测试 - split_v UT: `bash build.sh -u --opkernel_aicpu --ops=split_v --soc=ascend910b` - split_d UT: `bash build.sh -u --opkernel_aicpu --ops=split_d --soc=ascend910b` 4. GE IR验证 - 编包 → 装包 → 执行 GE IR用例流程通过 ## 文档更新 - 新增 conversion/split_d/README.md，说明split_d算子的设计文档 - docs/zh/op_list.md新增split_v, split_d aicpu算子支持说明 ## 类型标签 - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!2275	1 个月前

SplitD

产品支持情况

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

功能说明

算子功能：将张量沿指定维度split_dim平均拆分为num_split份更小的张量。与Split算子不同，split_dim作为属性而非输入提供。

参数说明

参数名	输入/输出/属性	描述	数据类型	数据格式
x	输入	需要切分的tensor。	FLOAT16、FLOAT、DOUBLE、INT8、INT16、INT32、INT64、UINT8、UINT16、UINT32、UINT64、BOOL	ND
split_dim	属性	指定沿其分割的维度。	INT	-
num_split	属性	指定要分割的tensor个数。	INT	-
y	输出	输出结果，动态输出，输出个数由num_split决定。	FLOAT16、FLOAT、DOUBLE、INT8、INT16、INT32、INT64、UINT8、UINT16、UINT32、UINT64、BOOL	ND

约束说明

split_dim必须在[0, value维度数)范围内，若为负数则从末尾开始计数
value在split_dim维度上的大小必须能被num_split整除
num_split必须大于等于1

调用说明

调用方式	样例代码	说明
图模式接口	test_geir_split_d	通过算子IR接口方式调用SplitD算子。