Star824

Star824

cann-robot精简Index算子tiling构建配置并统一op_host目录编译

cb8c2d18创建于 26 天前历史提交

文件	最后提交记录	最后更新时间
op_api	format cpp Co-authored-by: yang-di52<yangdi52@huawei.com> # message auto-generated for no-merge-commit merge: !6784 merge issue_fix into master format cpp Created-by: yang-di52 Commit-by: yang-di52 Merged-by: cann-robot Description: ## 描述批量刷新cpp代码格式 ## 关联的Issue [#3791](https://gitcode.com/cann/ops-nn/issues/3791) ## 测试 <!--描述进行了哪些测试来验证你的改动。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [x] 其他，请描述：代码格式化 ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [ ] AI辅助编写 See merge request: cann/ops-nn!6784	27 天前
op_graph	format cpp Co-authored-by: yang-di52<yangdi52@huawei.com> # message auto-generated for no-merge-commit merge: !6784 merge issue_fix into master format cpp Created-by: yang-di52 Commit-by: yang-di52 Merged-by: cann-robot Description: ## 描述批量刷新cpp代码格式 ## 关联的Issue [#3791](https://gitcode.com/cann/ops-nn/issues/3791) ## 测试 <!--描述进行了哪些测试来验证你的改动。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [x] 其他，请描述：代码格式化 ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [ ] AI辅助编写 See merge request: cann/ops-nn!6784	27 天前
op_host	精简Index算子tiling构建配置并统一op_host目录编译 Co-authored-by: tieyutong<tieyutong@huawei.com> # message auto-generated for no-merge-commit merge: !7072 merge index-tiling-fix-20260706 into master 精简Index算子tiling构建配置并统一op_host目录编译 Created-by: tieyutong Commit-by: tieyutong Merged-by: cann-robot Description: ## 描述根据评审意见，Index 算子的 tiling 实现同时支持 ascend950 和 ascend910b，不需要在 CMakeLists.txt 中通过 SUPPORT_COMPUTE_UNIT / SUPPORT_TILING_DIR 额外配置两个芯片到同一个 arch35 目录。本次修改： -移除 experimental/index/index/CMakeLists.txt 中冗余的 SUPPORT_COMPUTE_UNIT、SUPPORT_TILING_DIR 及对应参数。 -将 op_host/arch35/index_tiling.cpp 上移到 op_host/index_tiling.cpp，使其在两个芯片构建时统一参与编译。 -同步将 tests/ut/op_host/arch35/test_index_tiling.cpp 上移到 tests/ut/op_host/test_index_tiling.cpp，避免 tiling UT 被当前测试收集规则漏掉。 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> https://gitcode.com/cann/ops-nn/issues/3875 ## 测试已完成以下验证： clang-format --dry-run --Werror 通过。 bash build.sh -u --experimental --ops=index --ophost 通过，执行 8 tests from 2 test suites，全部通过。 bash scripts/ci/local_build.sh -u -f /tmp/index_pr_filelist.txt -j16： ophost ut 通过。 opapi ut 通过。 ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [x] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [x] AI辅助编写 See merge request: cann/ops-nn!7072	26 天前
op_kernel	format cpp Co-authored-by: yang-di52<yangdi52@huawei.com> # message auto-generated for no-merge-commit merge: !6784 merge issue_fix into master format cpp Created-by: yang-di52 Commit-by: yang-di52 Merged-by: cann-robot Description: ## 描述批量刷新cpp代码格式 ## 关联的Issue [#3791](https://gitcode.com/cann/ops-nn/issues/3791) ## 测试 <!--描述进行了哪些测试来验证你的改动。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [x] 其他，请描述：代码格式化 ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [ ] AI辅助编写 See merge request: cann/ops-nn!6784	27 天前
tests	精简Index算子tiling构建配置并统一op_host目录编译 Co-authored-by: tieyutong<tieyutong@huawei.com> # message auto-generated for no-merge-commit merge: !7072 merge index-tiling-fix-20260706 into master 精简Index算子tiling构建配置并统一op_host目录编译 Created-by: tieyutong Commit-by: tieyutong Merged-by: cann-robot Description: ## 描述根据评审意见，Index 算子的 tiling 实现同时支持 ascend950 和 ascend910b，不需要在 CMakeLists.txt 中通过 SUPPORT_COMPUTE_UNIT / SUPPORT_TILING_DIR 额外配置两个芯片到同一个 arch35 目录。本次修改： -移除 experimental/index/index/CMakeLists.txt 中冗余的 SUPPORT_COMPUTE_UNIT、SUPPORT_TILING_DIR 及对应参数。 -将 op_host/arch35/index_tiling.cpp 上移到 op_host/index_tiling.cpp，使其在两个芯片构建时统一参与编译。 -同步将 tests/ut/op_host/arch35/test_index_tiling.cpp 上移到 tests/ut/op_host/test_index_tiling.cpp，避免 tiling UT 被当前测试收集规则漏掉。 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> https://gitcode.com/cann/ops-nn/issues/3875 ## 测试已完成以下验证： clang-format --dry-run --Werror 通过。 bash build.sh -u --experimental --ops=index --ophost 通过，执行 8 tests from 2 test suites，全部通过。 bash scripts/ci/local_build.sh -u -f /tmp/index_pr_filelist.txt -j16： ophost ut 通过。 opapi ut 通过。 ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [x] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [x] AI辅助编写 See merge request: cann/ops-nn!7072	26 天前
CMakeLists.txt	精简Index算子tiling构建配置并统一op_host目录编译 Co-authored-by: tieyutong<tieyutong@huawei.com> # message auto-generated for no-merge-commit merge: !7072 merge index-tiling-fix-20260706 into master 精简Index算子tiling构建配置并统一op_host目录编译 Created-by: tieyutong Commit-by: tieyutong Merged-by: cann-robot Description: ## 描述根据评审意见，Index 算子的 tiling 实现同时支持 ascend950 和 ascend910b，不需要在 CMakeLists.txt 中通过 SUPPORT_COMPUTE_UNIT / SUPPORT_TILING_DIR 额外配置两个芯片到同一个 arch35 目录。本次修改： -移除 experimental/index/index/CMakeLists.txt 中冗余的 SUPPORT_COMPUTE_UNIT、SUPPORT_TILING_DIR 及对应参数。 -将 op_host/arch35/index_tiling.cpp 上移到 op_host/index_tiling.cpp，使其在两个芯片构建时统一参与编译。 -同步将 tests/ut/op_host/arch35/test_index_tiling.cpp 上移到 tests/ut/op_host/test_index_tiling.cpp，避免 tiling UT 被当前测试收集规则漏掉。 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> https://gitcode.com/cann/ops-nn/issues/3875 ## 测试已完成以下验证： clang-format --dry-run --Werror 通过。 bash build.sh -u --experimental --ops=index --ophost 通过，执行 8 tests from 2 test suites，全部通过。 bash scripts/ci/local_build.sh -u -f /tmp/index_pr_filelist.txt -j16： ophost ut 通过。 opapi ut 通过。 ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [x] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [x] AI辅助编写 See merge request: cann/ops-nn!7072	26 天前
README.md	新增index算子Ascend C实现 Co-authored-by: tieyutong<tieyutong@huawei.com> # message auto-generated for no-merge-commit merge: !6086 merge index-upload-20260615-1055 into master 新增index算子Ascend C实现 Created-by: tieyutong Commit-by: tieyutong Merged-by: cann-robot Description: ## 描述本 PR 新增 experimental Index 算子实现，提交路径为 `experimental/index/index`。主要改动包括： - 新增 Index 算子的工程配置、op_graph、op_host、op_api、op_kernel 及 UT 测试目录。 - 新增 `aclnnIndexGetWorkspaceSize` / `aclnnIndex` 接口，支持通过 aclnn 路径调用 Index 算子。 - 支持 `ascend910b` 和 `ascend950` AICore 二进制配置。 - 支持输入 `x` / 输出 `y` 数据类型：`FLOAT`、`DOUBLE`、`FLOAT16`、`BFLOAT16`、`INT16`、`INT32`、`INT64`、`INT8`、`UINT8`、`BOOL`。 - 普通 `INT32` / `INT64` 索引走 `IndexAiCore` 路径；BOOL 索引、索引数量超过输入 rank 等特殊路径走 `IndexAiCpu` 或直接返回。 ## 关联的Issue https://gitcode.com/cann/ops-nn/issues/3342 ## 测试 \| 类型 \| 条数 \| 通过率 / 结论 \| \|---\|---:\|---\| \| Smoke_Generalization \| 20 / 20 \| 100%，Pass \| \| Abnormal \| 3 / 3 \| 100%，Pass \| \| Sanitizer_Generalization \| 600 / 600 \| 100%，Pass \| \| Special \| 90 / 90 \| 100%，Pass \| \| SMK_over_5000w \| 1 / 1 \| 100%，Pass \| \| ACCU_O1 \| 82 / 82 \| 100%，Pass \| \| ACCU_over_5000w \| 24 / 24 \| 100%，Pass \| \| Perf_Generalization \| 20 / 20 \| device 性能通过率 100%，性能比0.9562，Pass \| \| Accu_Generalization \| 30000 / 30000 \| 100%，Pass \| ## 文档更新更新了 `experimental/index/index/README.md`。 ## 类型标签 <!-- [x] 表示选中 --> - [] Bug修复 - [x] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-nn!6086	29 天前

Index

产品支持情况

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

功能说明

算子功能：根据 indices 从输入张量 x 中取出指定位置的数据，生成输出张量 y。
接口形态：当前实验目录提供 aclnnIndexGetWorkspaceSize / aclnnIndex 接口，内部将输入和索引转为连续张量后执行 Index 计算。
执行路径：普通 INT32 / INT64 索引走 IndexAiCore；BOOL 索引、索引数量超过输入 rank 等特殊路径走 IndexAiCpu 或直接返回。

参数说明

参数名	输入/输出	描述	数据类型	数据格式	维度
x	输入	待索引的输入张量，对应 aclnn 接口中的 self。	FLOAT、DOUBLE、FLOAT16、BFLOAT16、INT16、INT32、INT64、INT8、UINT8、BOOL	ND	0-8
indexed_sizes	输入	内部元数据，记录被索引维度信息，由 aclnn 接口根据 self 和 indices 生成。	INT64	ND	1
indexed_strides	输入	内部元数据，记录输入张量 stride 或 BOOL 索引输出形状信息，由 aclnn 接口生成。	INT64	ND	1
indices	动态输入	索引张量列表。普通 AICore 路径支持 INT32、INT64；aclnn 参数检查接受 BOOL，BOOL 索引走 AICPU 路径。	INT32、INT64、BOOL	ND	0-8
y	输出	根据索引取出后的输出张量，数据类型需与 x 一致。	FLOAT、DOUBLE、FLOAT16、BFLOAT16、INT16、INT32、INT64、INT8、UINT8、BOOL	ND	0-8

约束说明

仅支持 ND 格式，不支持私有格式输入、索引或输出。
x、indices 中每个索引张量、y 的 rank 均不超过 8。
普通索引路径要求 indices 数量在 [1, x rank] 范围内，且各索引张量 shape 可广播；输出 shape 需与接口推导结果一致。
空 Tensor 和标量输入的单索引场景由 aclnn 接口做特殊处理。
AICore 二进制当前配置覆盖 ascend910b 和 ascend950。

调用说明

调用方式	接口/文件	说明
aclnn 调用	`op_api/aclnn_index_v2.h`（`aclnnIndexGetWorkspaceSize` / `aclnnIndex`）	当前目录未提供 `examples/test_aclnn_index.cpp`，可通过 aclnn 接口头文件和 ATK 用例进行验证。
Level0 内部调用	`op_api/experimental_index_l0.h`（`IndexAiCore` / `IndexAiCpu`）	aclnn 接口根据索引类型和形状选择 AICore 或 AICPU 路径。