文件最后提交记录最后更新时间
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_NDge::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!29726 天前
下一代strided_slice_grad等算子功能支持 Co-authored-by: qq_57796320<zhushuqin1@huawei.com> # message auto-generated for no-merge-commit merge: !569 merge master into master 下一代strided_slice_grad等算子功能支持 Created-by: qq_57796320 Commit-by: qq_57796320 Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动,包括改动的原因和所采取的方法。--> ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue,请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单,请在这里描述问题单单号。--> ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新,请在这里指出。例如:更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他,请描述: See merge request: cann/ops-math!5695 个月前
refactor: 拆分 tiling 公共头文件,消除 conversion 算子重复依赖 Co-authored-by: tianqiguang<tianqiguang@huawei.com> # message auto-generated for no-merge-commit merge: !2951 merge feature/conversion-dedup into master refactor: 拆分 tiling 公共头文件,消除 conversion 算子重复依赖 Created-by: tianqiguang Commit-by: tianqiguang Merged-by: cann-robot Description: ## 描述 将 tiling 基类和模板注册相关定义从 conversion 算子的公共头文件中拆分,消除各 conversion 算子的重复头文件依赖。 ### 改动原因 conversion 目录下 50+ 个算子各自直接引用 tiling_base.htiling_util.htiling_templates_registry.h: - tiling_base.h 包含完整 TilingBaseClass 实现(~240行),每个算子编译时都重复展开 - tiling_templates_registry.h 内混杂 math 算子的模板注册代码,不应被 conversion 算子依赖 ### 改动方法 1. 新建 common/inc/op_host/math_tiling_templates_registry.h:将 math 算子专用的模板注册宏和注册类从此文件中提供 2. 精简 tiling_base.h:改为仅转发引用 tiling_base_class.h 3. 批量更新 conversion 算子文件的 #include: - tiling_base.htiling_base_class.h - tiling_util.htiling_base_util.h - tiling_templates_registry.hmath_tiling_templates_registry.h 4. 批量更新命名空间引用: - Ops::Math::OpTiling::TilingBaseClassOps::Base::TilingBaseClass - Ops::Math::OpTiling::EnsureNotScalarOps::Base::EnsureNotScalar ## 关联的Issue - #1672 ## 测试 - 涉及的 conversion 算子二级冒烟测试已通过 ## 文档更新 无文档变更 ## 类型标签 - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [x] 其他,请描述:代码重构,消除重复头文件依赖 See merge request: cann/ops-math!29518 天前
下一代strided_slice_grad等算子功能支持 Co-authored-by: qq_57796320<zhushuqin1@huawei.com> # message auto-generated for no-merge-commit merge: !569 merge master into master 下一代strided_slice_grad等算子功能支持 Created-by: qq_57796320 Commit-by: qq_57796320 Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动,包括改动的原因和所采取的方法。--> ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue,请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单,请在这里描述问题单单号。--> ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新,请在这里指出。例如:更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他,请描述: See merge request: cann/ops-math!5695 个月前
unpack/tile/cosh ge ir用例dt_double支持 Co-authored-by: cann-robot<cann@cann.team> # message auto-generated for no-merge-commit merge: !2882 merge dev_extra into master unpack/tile/cosh ge ir用例dt_double支持 Created-by: sujunwei3 Commit-by: cann-robot Merged-by: cann-robot Description: ## 描述 为 unpack/tile/cosh 三个算子的 GE IR 用例添加 DT_DOUBLE 数据类型支持,用于触发 AICPU fallback 验证。 主要改动: 1. **unpack GE IR 用例** (conversion/unpack/examples/test_geir_unpack.cpp) - 删除不存在的 experiment_ops.h 依赖 - 重构 CreateOppInGraph 函数,使用正确的 Data operator 方式 - 添加 GetDataTypeSizeGenOnesData 对 DT_DOUBLE 的支持 - 使用 DT_DOUBLE 作为默认 dtype,验证 AICPU 流程 2. **tile GE IR 用例** (math/tile/examples/test_geir_tile.cpp) - 新增 ADD_INPUT_DOUBLE 宏用于 DT_DOUBLE 输入处理 - 新增 GenDataDouble 函数生成 double 类型数据 - 添加 DT_DOUBLE 到 GetDataTypeSizePrintOutputData - 保留原始 DT_INT32 默认 dtype,DT_DOUBLE 支持可用于手动切换验证 AICPU 3. **cosh GE IR 用例** (math/cosh/examples/test_geir_cosh.cpp) - 添加 DT_DOUBLE 数据类型支持 - 保留原始 DT_FLOAT 默认 dtype 4. **unpack AICPU kernel** (conversion/unpack/op_kernel_aicpu/unpack_aicpu.cpp) - 小幅调整(+1/-1) ## 关联的Issue 暂无 ## 测试 已验证以下测试场景: 1. **编译构建测试** - 三个算子的 GE IR 用例编译通过 - 无编译错误和警告 2. **GE IR 流程验证** - tile 算子:DT_INT32 默认 dtype 执行成功 - cosh 算子:DT_FLOAT 默认 dtype 执行成功 - unpack 算子:DT_DOUBLE 执行成功,自动触发 AICPU fallback(CUSTAICPUKernel) 3. **完整流程验证** - 编包 → 装包 → GE IR 执行全流程通过 - 验证了 DT_DOUBLE 可触发自定义 AICPU kernel 执行 ## 文档更新 无 ## 类型标签 - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他,请描述:测试用例增强 See merge request: cann/ops-math!288213 天前
Unpack aicpu Co-authored-by: shikai-1<shikai4@huawei.com> # message auto-generated for no-merge-commit merge: !2072 merge master into master Unpack aicpu Created-by: shikai-1 Commit-by: shikai-1 Merged-by: cann-robot Description: ## 描述 AICPU算子Unpack算子开源 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue,请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单,请在这里描述问题单单号。--> https://gitcode.com/cann/ops-math/issues/1176 ## 测试 增加了UT ## 文档更新 <!--如果这个PR包含文档的更新,请在这里指出。例如:更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他,请描述: See merge request: cann/ops-math!20721 个月前
Arch编码更新 Co-authored-by: liulinxiang<liulinxiang1@huawei.com> # message auto-generated for no-merge-commit merge: !992 merge master1 into master Arch编码更新 Created-by: liulinxiang Commit-by: liulinxiang Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动,包括改动的原因和所采取的方法。--> ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue,请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单,请在这里描述问题单单号。--> ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新,请在这里指出。例如:更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他,请描述: See merge request: cann/ops-math!9924 个月前
doc Tools扫描错误修改 Co-authored-by: gitee-yanglulu<yanglulul@h-partners.com> # message auto-generated for no-merge-commit merge: !1899 merge master into master doc Tools扫描错误修改 Created-by: gitee-yanglulu Commit-by: gitee-yanglulu Merged-by: cann-robot Description: doc Tools扫描错误修改 See merge request: cann/ops-math!18992 个月前
README.md

Unpack

产品支持情况

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

功能说明

  • 算子功能:将张量沿着某一维度拆分为多个子张量。

参数说明

参数名 输入/输出/属性 描述 数据类型 数据格式
x 输入 输入tensor INT64、INT32、UINT32、FLOAT、FLOAT16、INT8、UINT8、BFLOAT16、INT16、UINT16、BOOL ND
num 输入属性 拆分的数量 INT -
axis 输入属性 拆分的维度 INT -
y 输出 输入张量在指定维度上的大小 INT64、INT32、UINT32、FLOAT、FLOAT16、INT8、UINT8、BFLOAT16、INT16、UINT16、BOOL ND

约束说明

调用说明

调用方式 调用样例 说明
图模式调用 test_geir_unpack 通过算子IR构图方式调用Unpack算子。