Star1241

Star1241

cann-robotadd aicpu:tile_with_axis&topkpqdistance

997507c9创建于 16 天前历史提交

文件	最后提交记录	最后更新时间
examples	add aicpu:tile_with_axis&topkpqdistance Co-authored-by: sujunwei3<sujunwei3@huawei.com> # message auto-generated for no-merge-commit merge: !4174 merge dev into master add aicpu:tile_with_axis&topkpqdistance Created-by: sujunwei3 Commit-by: sujunwei3 Merged-by: cann-robot Description: ## 描述新增四个 AI CPU 算子： 1. tile_with_axis（conversion）：沿指定维度复制输入 Tensor 数据，扩展输出 Tensor。包含 AI Core 和 AI CPU 双实现。 2. top_k_pq_distance（math）：基于分组极值剪枝的 TopK 距离计算，用于向量检索场景。支持动态多组输入。 3. top_k_v2_d（math）：沿指定维度找出最大或最小的 k 个元素及其索引，兼容 TensorFlow TopKV2，包含 infershape 实现。 4. zeta（math）：计算 Hurwitz zeta 函数 ζ(x, q) = Σ(n=0→∞) 1/(q+n)^x。每个算子包含：op_kernel_aicpu 实现、op_graph 构图定义、examples 示例、UT 单元测试、README 文档。 ## 关联的Issue #2294 ## 测试根据代码变更，测试场景如下： 1. tile_with_axis 单元测试 + GE IR 验证 - 执行 `bash build.sh -u --opkernel_aicpu --ops=tile_with_axis --soc=ascend910b`，17 个 UT 全部通过 - 执行编包 + GE IR 运行，`execute samples success`，输出 bin 文件正常 2. top_k_pq_distance 单元测试 - 执行 `bash build.sh -u --opkernel_aicpu --ops=top_k_pq_distance --soc=ascend910b`，5 个 UT 全部通过 3. top_k_v2_d 单元测试 + GE IR 验证 - 执行 `bash build.sh -u --opkernel_aicpu --ops=top_k_v2_d --soc=ascend910b`，UT 全部通过 - 执行编包 + GE IR 运行，验证通过 4. zeta 单元测试 + GE IR 验证 - 执行 `bash build.sh -u --opkernel_aicpu --ops=zeta --soc=ascend910b`，8 个 UT 全部通过 - 执行编包 + GE IR 运行，`execute samples success`，输出 bin 文件正常 ## 文档更新 - 新增 4 个算子 README.md（tile_with_axis、top_k_pq_distance、top_k_v2_d、zeta） - 更新 docs/zh/op_list.md，新增 4 个算子条目 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!4174	16 天前
op_graph	add aicpu:tile_with_axis&topkpqdistance Co-authored-by: sujunwei3<sujunwei3@huawei.com> # message auto-generated for no-merge-commit merge: !4174 merge dev into master add aicpu:tile_with_axis&topkpqdistance Created-by: sujunwei3 Commit-by: sujunwei3 Merged-by: cann-robot Description: ## 描述新增四个 AI CPU 算子： 1. tile_with_axis（conversion）：沿指定维度复制输入 Tensor 数据，扩展输出 Tensor。包含 AI Core 和 AI CPU 双实现。 2. top_k_pq_distance（math）：基于分组极值剪枝的 TopK 距离计算，用于向量检索场景。支持动态多组输入。 3. top_k_v2_d（math）：沿指定维度找出最大或最小的 k 个元素及其索引，兼容 TensorFlow TopKV2，包含 infershape 实现。 4. zeta（math）：计算 Hurwitz zeta 函数 ζ(x, q) = Σ(n=0→∞) 1/(q+n)^x。每个算子包含：op_kernel_aicpu 实现、op_graph 构图定义、examples 示例、UT 单元测试、README 文档。 ## 关联的Issue #2294 ## 测试根据代码变更，测试场景如下： 1. tile_with_axis 单元测试 + GE IR 验证 - 执行 `bash build.sh -u --opkernel_aicpu --ops=tile_with_axis --soc=ascend910b`，17 个 UT 全部通过 - 执行编包 + GE IR 运行，`execute samples success`，输出 bin 文件正常 2. top_k_pq_distance 单元测试 - 执行 `bash build.sh -u --opkernel_aicpu --ops=top_k_pq_distance --soc=ascend910b`，5 个 UT 全部通过 3. top_k_v2_d 单元测试 + GE IR 验证 - 执行 `bash build.sh -u --opkernel_aicpu --ops=top_k_v2_d --soc=ascend910b`，UT 全部通过 - 执行编包 + GE IR 运行，验证通过 4. zeta 单元测试 + GE IR 验证 - 执行 `bash build.sh -u --opkernel_aicpu --ops=zeta --soc=ascend910b`，8 个 UT 全部通过 - 执行编包 + GE IR 运行，`execute samples success`，输出 bin 文件正常 ## 文档更新 - 新增 4 个算子 README.md（tile_with_axis、top_k_pq_distance、top_k_v2_d、zeta） - 更新 docs/zh/op_list.md，新增 4 个算子条目 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!4174	16 天前
op_host	refactor(tile_with_axis) Co-authored-by: onanfield<tianshang@huawei.com> # message auto-generated for no-merge-commit merge: !3486 merge feat/tile_with_axis into master refactor(tile_with_axis) Created-by: onanfield Commit-by: onanfield Merged-by: cann-robot Description: ## 描述修复 tilewithaxis算子检视意见 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [x ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!3486	1 个月前
op_kernel	refactor(tile_with_axis) Co-authored-by: onanfield<tianshang@huawei.com> # message auto-generated for no-merge-commit merge: !3486 merge feat/tile_with_axis into master refactor(tile_with_axis) Created-by: onanfield Commit-by: onanfield Merged-by: cann-robot Description: ## 描述修复 tilewithaxis算子检视意见 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [x ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!3486	1 个月前
op_kernel_aicpu	add aicpu:tile_with_axis&topkpqdistance Co-authored-by: sujunwei3<sujunwei3@huawei.com> # message auto-generated for no-merge-commit merge: !4174 merge dev into master add aicpu:tile_with_axis&topkpqdistance Created-by: sujunwei3 Commit-by: sujunwei3 Merged-by: cann-robot Description: ## 描述新增四个 AI CPU 算子： 1. tile_with_axis（conversion）：沿指定维度复制输入 Tensor 数据，扩展输出 Tensor。包含 AI Core 和 AI CPU 双实现。 2. top_k_pq_distance（math）：基于分组极值剪枝的 TopK 距离计算，用于向量检索场景。支持动态多组输入。 3. top_k_v2_d（math）：沿指定维度找出最大或最小的 k 个元素及其索引，兼容 TensorFlow TopKV2，包含 infershape 实现。 4. zeta（math）：计算 Hurwitz zeta 函数 ζ(x, q) = Σ(n=0→∞) 1/(q+n)^x。每个算子包含：op_kernel_aicpu 实现、op_graph 构图定义、examples 示例、UT 单元测试、README 文档。 ## 关联的Issue #2294 ## 测试根据代码变更，测试场景如下： 1. tile_with_axis 单元测试 + GE IR 验证 - 执行 `bash build.sh -u --opkernel_aicpu --ops=tile_with_axis --soc=ascend910b`，17 个 UT 全部通过 - 执行编包 + GE IR 运行，`execute samples success`，输出 bin 文件正常 2. top_k_pq_distance 单元测试 - 执行 `bash build.sh -u --opkernel_aicpu --ops=top_k_pq_distance --soc=ascend910b`，5 个 UT 全部通过 3. top_k_v2_d 单元测试 + GE IR 验证 - 执行 `bash build.sh -u --opkernel_aicpu --ops=top_k_v2_d --soc=ascend910b`，UT 全部通过 - 执行编包 + GE IR 运行，验证通过 4. zeta 单元测试 + GE IR 验证 - 执行 `bash build.sh -u --opkernel_aicpu --ops=zeta --soc=ascend910b`，8 个 UT 全部通过 - 执行编包 + GE IR 运行，`execute samples success`，输出 bin 文件正常 ## 文档更新 - 新增 4 个算子 README.md（tile_with_axis、top_k_pq_distance、top_k_v2_d、zeta） - 更新 docs/zh/op_list.md，新增 4 个算子条目 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!4174	16 天前
tests	add aicpu:tile_with_axis&topkpqdistance Co-authored-by: sujunwei3<sujunwei3@huawei.com> # message auto-generated for no-merge-commit merge: !4174 merge dev into master add aicpu:tile_with_axis&topkpqdistance Created-by: sujunwei3 Commit-by: sujunwei3 Merged-by: cann-robot Description: ## 描述新增四个 AI CPU 算子： 1. tile_with_axis（conversion）：沿指定维度复制输入 Tensor 数据，扩展输出 Tensor。包含 AI Core 和 AI CPU 双实现。 2. top_k_pq_distance（math）：基于分组极值剪枝的 TopK 距离计算，用于向量检索场景。支持动态多组输入。 3. top_k_v2_d（math）：沿指定维度找出最大或最小的 k 个元素及其索引，兼容 TensorFlow TopKV2，包含 infershape 实现。 4. zeta（math）：计算 Hurwitz zeta 函数 ζ(x, q) = Σ(n=0→∞) 1/(q+n)^x。每个算子包含：op_kernel_aicpu 实现、op_graph 构图定义、examples 示例、UT 单元测试、README 文档。 ## 关联的Issue #2294 ## 测试根据代码变更，测试场景如下： 1. tile_with_axis 单元测试 + GE IR 验证 - 执行 `bash build.sh -u --opkernel_aicpu --ops=tile_with_axis --soc=ascend910b`，17 个 UT 全部通过 - 执行编包 + GE IR 运行，`execute samples success`，输出 bin 文件正常 2. top_k_pq_distance 单元测试 - 执行 `bash build.sh -u --opkernel_aicpu --ops=top_k_pq_distance --soc=ascend910b`，5 个 UT 全部通过 3. top_k_v2_d 单元测试 + GE IR 验证 - 执行 `bash build.sh -u --opkernel_aicpu --ops=top_k_v2_d --soc=ascend910b`，UT 全部通过 - 执行编包 + GE IR 运行，验证通过 4. zeta 单元测试 + GE IR 验证 - 执行 `bash build.sh -u --opkernel_aicpu --ops=zeta --soc=ascend910b`，8 个 UT 全部通过 - 执行编包 + GE IR 运行，`execute samples success`，输出 bin 文件正常 ## 文档更新 - 新增 4 个算子 README.md（tile_with_axis、top_k_pq_distance、top_k_v2_d、zeta） - 更新 docs/zh/op_list.md，新增 4 个算子条目 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!4174	16 天前
CMakeLists.txt	feat(tile_with_axis): add tile_with_axis operator with arch35 support Co-authored-by: onanfield<tianshang@huawei.com> # message auto-generated for no-merge-commit merge: !3374 merge feat/tile_with_axis into master feat(tile_with_axis): add tile_with_axis operator with arch35 support Created-by: onanfield Commit-by: onanfield Merged-by: cann-robot Description: ## 描述 tile_with_axis算子支持A5芯片 ## 关联的Issue 无 ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!3374	1 个月前
README.md	add aicpu:tile_with_axis&topkpqdistance Co-authored-by: sujunwei3<sujunwei3@huawei.com> # message auto-generated for no-merge-commit merge: !4174 merge dev into master add aicpu:tile_with_axis&topkpqdistance Created-by: sujunwei3 Commit-by: sujunwei3 Merged-by: cann-robot Description: ## 描述新增四个 AI CPU 算子： 1. tile_with_axis（conversion）：沿指定维度复制输入 Tensor 数据，扩展输出 Tensor。包含 AI Core 和 AI CPU 双实现。 2. top_k_pq_distance（math）：基于分组极值剪枝的 TopK 距离计算，用于向量检索场景。支持动态多组输入。 3. top_k_v2_d（math）：沿指定维度找出最大或最小的 k 个元素及其索引，兼容 TensorFlow TopKV2，包含 infershape 实现。 4. zeta（math）：计算 Hurwitz zeta 函数 ζ(x, q) = Σ(n=0→∞) 1/(q+n)^x。每个算子包含：op_kernel_aicpu 实现、op_graph 构图定义、examples 示例、UT 单元测试、README 文档。 ## 关联的Issue #2294 ## 测试根据代码变更，测试场景如下： 1. tile_with_axis 单元测试 + GE IR 验证 - 执行 `bash build.sh -u --opkernel_aicpu --ops=tile_with_axis --soc=ascend910b`，17 个 UT 全部通过 - 执行编包 + GE IR 运行，`execute samples success`，输出 bin 文件正常 2. top_k_pq_distance 单元测试 - 执行 `bash build.sh -u --opkernel_aicpu --ops=top_k_pq_distance --soc=ascend910b`，5 个 UT 全部通过 3. top_k_v2_d 单元测试 + GE IR 验证 - 执行 `bash build.sh -u --opkernel_aicpu --ops=top_k_v2_d --soc=ascend910b`，UT 全部通过 - 执行编包 + GE IR 运行，验证通过 4. zeta 单元测试 + GE IR 验证 - 执行 `bash build.sh -u --opkernel_aicpu --ops=zeta --soc=ascend910b`，8 个 UT 全部通过 - 执行编包 + GE IR 运行，`execute samples success`，输出 bin 文件正常 ## 文档更新 - 新增 4 个算子 README.md（tile_with_axis、top_k_pq_distance、top_k_v2_d、zeta） - 更新 docs/zh/op_list.md，新增 4 个算子条目 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!4174	16 天前

TileWithAxis

产品支持情况

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

功能说明

算子功能：沿指定维度复制输入Tensor数据，扩展输出Tensor。
计算示例：
- 输入 x = [[[1, 2], [3, 4], [5, 6]], [[7, 8], [9, 10], [11, 12]]]，shape = (2, 3, 2)
- axis = 1，tiles = 2
- 输出 y = [[[1, 2], [3, 4], [5, 6], [1, 2], [3, 4], [5, 6]], [[7, 8], [9, 10], [11, 12], [7, 8], [9, 10], [11, 12]]]，shape = (2, 6, 2)
计算公式：输出 y 的 shape 与输入 x 相同，仅 axis 维度变为 x.shape[axis] * tiles。

参数说明

参数名	输入/输出/属性	描述	数据类型	数据格式
x	输入	输入张量	FLOAT16、FLOAT、INT8、INT16、INT32、INT64、UINT8、UINT16、UINT32、UINT64	ND
axis	属性（可选）	指定复制的维度，默认为1	INT	-
tiles	属性（必选）	复制次数，必须大于0	INT	-
y	输出	输出张量，与x具有相同的数据类型和格式	与x一致	ND

约束说明

axis 必须在输入Tensor的维度范围内（支持负数索引，如 axis=-1 表示最后一维）。
tiles 必须大于0。
输出Tensor在 axis 维度的大小必须等于输入Tensor在 axis 维度的大小乘以 tiles。
输入Tensor维度不超过8维。

调用说明

调用方式	样例代码	说明
图模式调用	test_geir_tile_with_axis	通过算子IR构图方式调用TileWithAxis算子。