Star0
0
代码
代码
Issues
Pull Requests
Actions
项目讨论
Wiki
分析
Star0
0
文件最后提交记录最后更新时间
docs
删除查看源码 Co-authored-by: y60124828<yanglou1@h-partners.com> # message auto-generated for no-merge-commit merge: !12004 merge master into master 删除查看源码 Created-by: yanglu-1 Commit-by: y60124828 Merged-by: cann-robot Description: ## 描述 删除查看源码 ## 关联的Issue 关联Issue [#6694](https://gitcode.com/cann/ops-nn/issues/6694) ## 测试 无需测试 ## 文档更新 删除文档中的查看源码链接 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [ ] AI辅助编写 See merge request: cann/ops-nn!12004
8 天前
examples
算子日志可读性与代码规范性的优化 Co-authored-by: xingtaowu<wuxingtao3@h-partners.com> # message auto-generated for no-merge-commit merge: !9376 merge fix_nn_log into master 算子日志可读性与代码规范性的优化 Created-by: xingtaowu Commit-by: xingtaowu Merged-by: cann-robot Description: ## 描述 批量修正日志/报错信息中的英文拼写错误与语法问题，并补充更明确的参数越界诊断信息，同时修正示例测试程序中的日志级别与代码格式问题。改动不涉及业务逻辑与算法流程变化，均为日志可读性与代码规范性的优化。 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> ## 关联的Issue https://gitcode.com/cann/ops-nn/issues/5291 <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> ## 测试 冒烟通过 <!--描述进行了哪些测试来验证你的改动。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ x] 其他，请描述：算子日志可读性与代码规范性的优化 ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [ ] AI辅助编写 See merge request: cann/ops-nn!9376
1 个月前
op_graph
代码同步 Co-authored-by: yang-di52<yangdi52@huawei.com> # message auto-generated for no-merge-commit merge: !450 merge master into master 代码同步 Created-by: yang-di52 Commit-by: yang-di52 Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-nn!450
9 个月前
op_host
feat: cross_entropy_sum_exp_and_index_logit 与 repeat_interleave_grad 新增Ascend960平台支持 Co-authored-by: wkyan<yanwenkai@huawei.com> # message auto-generated for no-merge-commit merge: !10813 merge v121 into master feat: cross_entropy_sum_exp_and_index_logit 与 repeat_interleave_grad 新增Ascend960平台支持 Created-by: jzj007 Commit-by: wkyan Merged-by: cann-robot Description: ## 描述 为 cross_entropy_sum_exp_and_index_logit 与 repeat_interleave_grad 两个算子新增 ascend960dt（arch35）芯片平台支持，使其可在 ascend960dt 上完成算子定义注册、host/kernel 编译与二进制 kernel 配置生成。本次改动为平台适配，未修改 kernel 计算逻辑，复用 arch35 tiling 与 kernel 实现。 ### cross_entropy_sum_exp_and_index_logit 1. loss/cross_entropy_sum_exp_and_index_logit/CMakeLists.txt：SUPPORT_COMPUTE_UNIT 新增 ascend960dt，SUPPORT_TILING_DIR 追加 arch35，复用 arch35 tiling。 2. op_host/cross_entropy_sum_exp_and_index_logit_def.cpp：新增 this->AICore().AddConfig("ascend960dt", config_950)。 3. op_kernel/CMakeLists.txt：add_kernel_sources 的 COMPUTE_UNITS 新增 ascend960dt，复用 arch35/cross_entropy_sum_exp_and_index_logit.cpp。 4. 新增 op_host/config/ascend960dt/cross_entropy_sum_exp_and_index_logit_binary.json，含 float32、bfloat16 两套二进制 kernel 配置（输入 vocab_parallel_logits/target/global_logits_max，输出 predicted_logits/sum_exp_logits/exp_logits/target_offset/target_mask，含 vocab_start_index/vocab_end_index 属性）。 ### repeat_interleave_grad 1. index/repeat_interleave_grad/CMakeLists.txt：SUPPORT_COMPUTE_UNIT 新增 ascend960dt，SUPPORT_TILING_DIR 追加 arch35；add_kernel_sources 的 COMPUTE_UNITS 新增 ascend960dt。 2. op_host/repeat_interleave_grad_def.cpp：新增 this->AICore().AddConfig("ascend960dt", config_91095)。 3. 新增 op_host/config/ascend960dt/repeat_interleave_grad_binary.json，含 float16/bfloat16/float32（repeats 支持 int32/int64）共 6 套二进制 kernel 配置。 4. scripts/kernel/binary_config/ascendc_config.json：RepeatInterleaveGrad 的 compute_units 新增 ascend960dt。 ## 关联的Issue - 关联 Issue：https://gitcode.com/cann/ops-nn/issues/6767 ## 测试 本次改动为芯片平台适配（CMake 编译配置、算子定义注册、binary 配置），未修改 kernel 计算逻辑，复用 arch35 tiling 与 kernel 实现。需在 ascend960dt 环境完成编译及算子泛化、二级冒烟验证。 ## 文档更新 无 ## 类型标签 - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： ## AI/Agent生成声明 - [ ] AI辅助编写 See merge request: cann/ops-nn!10813
4 天前
op_kernel
重命名RIG算子中的重复文件名 platform.h Co-authored-by: wow5522<wuao1@huawei.com> # message auto-generated for no-merge-commit merge: !10798 merge master into master 重命名RIG算子中的重复文件名 platform.h Created-by: wow5523 Commit-by: wow5522 Merged-by: cann-robot Description: ## 描述 把repeat_interleave_grad算子中的文件名 platform.h改成repeat_interleave_grad_platform.h <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> ## 关联的Issue https://gitcode.com/cann/ops-nn/issues/6030 <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> ## 测试 <!--描述进行了哪些测试来验证你的改动。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [x] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [ ] AI辅助编写 See merge request: cann/ops-nn!10798
1 个月前
tests
补充 repeat_interleave_grad tiling arch35 UT 用例 Co-authored-by: andong_hw<andong2@huawei.com> # message auto-generated for no-merge-commit merge: !8426 merge rep_interl_grad into master 补充 repeat_interleave_grad tiling arch35 UT 用例 Created-by: andong_hw Commit-by: andong_hw Merged-by: cann-robot Description: ## 关联 Issue Closes #4649 ## 背景 repeat_interleave_grad_tiling_arch35.cpp 覆盖率仅 74.41%（211 行）。现有测试未覆盖 BLOCK_SPLIT_MN、BLOCK_SPLIT_R tiling 策略、空 tensor 等场景。 ## 修改内容 新增 3 个 UT 用例： - repeat_interleave_grad_tiling_block_split_mn：lenM*lenN >= coreNum，覆盖 BLOCK_SPLIT_MN 策略 - repeat_interleave_grad_tiling_block_split_r：lenM*lenRepeat >= coreNum，覆盖 BLOCK_SPLIT_R 策略 - repeat_interleave_grad_tiling_empty_tensor：空 tensor，覆盖空 tensor 分支 ## 验证 - 编译通过 - 本地环境 tiling 测试存在 platform mock segfault 问题（非代码问题），需 CI 环境验证 See merge request: cann/ops-nn!8426
2 个月前
CMakeLists.txt
feat: cross_entropy_sum_exp_and_index_logit 与 repeat_interleave_grad 新增Ascend960平台支持 Co-authored-by: wkyan<yanwenkai@huawei.com> # message auto-generated for no-merge-commit merge: !10813 merge v121 into master feat: cross_entropy_sum_exp_and_index_logit 与 repeat_interleave_grad 新增Ascend960平台支持 Created-by: jzj007 Commit-by: wkyan Merged-by: cann-robot Description: ## 描述 为 cross_entropy_sum_exp_and_index_logit 与 repeat_interleave_grad 两个算子新增 ascend960dt（arch35）芯片平台支持，使其可在 ascend960dt 上完成算子定义注册、host/kernel 编译与二进制 kernel 配置生成。本次改动为平台适配，未修改 kernel 计算逻辑，复用 arch35 tiling 与 kernel 实现。 ### cross_entropy_sum_exp_and_index_logit 1. loss/cross_entropy_sum_exp_and_index_logit/CMakeLists.txt：SUPPORT_COMPUTE_UNIT 新增 ascend960dt，SUPPORT_TILING_DIR 追加 arch35，复用 arch35 tiling。 2. op_host/cross_entropy_sum_exp_and_index_logit_def.cpp：新增 this->AICore().AddConfig("ascend960dt", config_950)。 3. op_kernel/CMakeLists.txt：add_kernel_sources 的 COMPUTE_UNITS 新增 ascend960dt，复用 arch35/cross_entropy_sum_exp_and_index_logit.cpp。 4. 新增 op_host/config/ascend960dt/cross_entropy_sum_exp_and_index_logit_binary.json，含 float32、bfloat16 两套二进制 kernel 配置（输入 vocab_parallel_logits/target/global_logits_max，输出 predicted_logits/sum_exp_logits/exp_logits/target_offset/target_mask，含 vocab_start_index/vocab_end_index 属性）。 ### repeat_interleave_grad 1. index/repeat_interleave_grad/CMakeLists.txt：SUPPORT_COMPUTE_UNIT 新增 ascend960dt，SUPPORT_TILING_DIR 追加 arch35；add_kernel_sources 的 COMPUTE_UNITS 新增 ascend960dt。 2. op_host/repeat_interleave_grad_def.cpp：新增 this->AICore().AddConfig("ascend960dt", config_91095)。 3. 新增 op_host/config/ascend960dt/repeat_interleave_grad_binary.json，含 float16/bfloat16/float32（repeats 支持 int32/int64）共 6 套二进制 kernel 配置。 4. scripts/kernel/binary_config/ascendc_config.json：RepeatInterleaveGrad 的 compute_units 新增 ascend960dt。 ## 关联的Issue - 关联 Issue：https://gitcode.com/cann/ops-nn/issues/6767 ## 测试 本次改动为芯片平台适配（CMake 编译配置、算子定义注册、binary 配置），未修改 kernel 计算逻辑，复用 arch35 tiling 与 kernel 实现。需在 ascend960dt 环境完成编译及算子泛化、二级冒烟验证。 ## 文档更新 无 ## 类型标签 - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： ## AI/Agent生成声明 - [ ] AI辅助编写 See merge request: cann/ops-nn!10813
4 天前
README.md
docs(repeat_interleave_grad): 修正输出数据类型说明 Co-authored-by: yangziqi_007<yangziqi4@huawei.com> # message auto-generated for no-merge-commit merge: !11835 merge woey/issue-6413-index-readme-dtypes into master docs(repeat_interleave_grad): 修正输出数据类型说明 Created-by: yangziqi_007 Commit-by: yangziqi_007 Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> docs(repeat_interleave_grad): 修正输出数据类型说明 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> 关联Issue [#6413](https://gitcode.com/cann/ops-nn/issues/6413) ## 测试 <!--描述进行了哪些测试来验证你的改动。--> 无 ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> index/repeat_interleave_grad/README.md ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [ ] AI辅助编写 See merge request: cann/ops-nn!11835
10 天前
README

RepeatInterleaveGrad

产品支持情况

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

功能说明

  • 算子功能：算子RepeatInterleave的反向,将y_grad tensor的axis维度按repeats进行ReduceSum。

参数说明

参数名 输入/输出/属性 描述 数据类型 数据格式
y_grad 输入 表示待被ReduceSum的输入tensor。 FLOAT、FLOAT16、BFLOAT16 ND
repeats 输入 表示重复的次数。 INT32、INT64 ND
x_grad 输出 表示ReduceSum完成的输出tensor。 FLOAT、FLOAT16、BFLOAT16 ND
axis 可选属性
  • 表示ReduceSum作用的维度。取值范围为[-n, n)，其中n为y_grad的维度。
  • 默认值为-1。
 int64_t -

约束说明

无

调用说明

调用方式 调用样例 说明
aclnn调用 test_aclnn_repeat_interleave_grad 通过aclnnRepeatInterleaveGrad接口方式调用RepeatInterleaveGrad算子。
图模式调用 test_geir_repeat_interleave_grad 通过算子IR构图方式调用RepeatInterleaveGrad算子。