1d50964b创建于 23 天前历史提交

文件	最后提交记录	最后更新时间
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	2 个月前
op_graph	math仓算子添加隔离宏 Co-authored-by: ayuemoon<15802951036@163.com> # message auto-generated for no-merge-commit merge: !5280 merge proto_reg_op_modify into master math仓算子添加隔离宏 Created-by: ayuemoon Commit-by: ayuemoon Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> 为 CalcBucketsLimitAndOffset、CaseCondition、ConjugateTranspose、CompareAndBitpack、Conj、GetDynamicDims、InplaceTopKDistance、ProdEnvMatACalcRij、TopKPQDistance、TopKPQDistanceV2、TopKV2D、Zeta 算子添加隔离宏，防止原型重复定义。 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> 无 ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> 无 ## 类型标签 <!-- [x] 表示选中 --> - [x] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!5280	23 天前
op_host	add infershape for top_k_pq_distance&zeta Co-authored-by: sujunwei3<sujunwei3@huawei.com> # message auto-generated for no-merge-commit merge: !4342 merge dev into master add infershape for top_k_pq_distance&zeta Created-by: sujunwei3 Commit-by: sujunwei3 Merged-by: cann-robot Description: ## 描述为 top_k_pq_distance 与 zeta 算子新增 host 侧 InferShape 实现，并将 top_k_pq_distance、top_k_v2_d、zeta 三个 AICPU 算子的注册宏从 `REGISTER_CPU_KERNEL` 迁移到 `OPS_MATH_REGISTER_CPU_KERNELV2`，同时在对应 CMakeLists.txt 中开启 `HOSTCPU TRUE` 以编译 host 侧 infershape 模块。主要改动： - 新增 `math/top_k_pq_distance/op_host/top_k_pq_distance_infershape.cpp`：三个输出 topk_distance/topk_ivf/topk_index 的 shape 均为 `{k}`（k 来自属性），dtype 分别继承自 pq_distance[0]/pq_ivf[0]/pq_index[0] 的动态输入。 - 新增 `math/zeta/op_host/zeta_infershape.cpp`：输出 z 的 shape 为 x 与 q 的广播结果（复用 `Ops::Base::InferShape4Broadcast`），dtype 继承自 x；并实现 InferShapeRange 以支持动态 shape range 场景。 - 修改 `math/{top_k_pq_distance,top_k_v2_d,zeta}/CMakeLists.txt`：`add_all_modules_sources(... HOSTCPU TRUE)`，启用 hostcpu infershape 编译。 - 修改 `math/{top_k_pq_distance,top_k_v2_d,zeta}/op_kernel_aicpu/_aicpu.cpp`：`REGISTER_CPU_KERNEL` → `OPS_MATH_REGISTER_CPU_KERNELV2`，引入 `aicpu/math_aicpu_register.h`。 - 更新 `docs/zh/op_list.md`：top_k_pq_distance、zeta 的 hostcpu infershape 支持状态由 × 更新为 √。 ## 关联的Issue #2472 https://gitcode.com/cann/ops-math/issues/2472 ## 测试根据代码变更，测试场景如下： 1. 编译构建测试* - 执行整仓编译 `bash build.sh --pkg --jit -j16`，验证三个算子 CMakeLists.txt 的 `HOSTCPU TRUE` 配置变更正确，hostcpu infershape 模块能正常编译产出。 2. InferShape 功能验证 - top_k_pq_distance：验证三个输出 topk_distance/topk_ivf/topk_index 的 shape 均为 `{k}`，dtype 分别与 pq_distance[0]/pq_ivf[0]/pq_index[0] 一致。 - zeta：验证输出 z 的 shape 为 x 与 q 广播后的 shape，dtype 与 x 一致；覆盖静态 shape 与动态 shape range 两种场景（含 0/1/-1 的 range 用例）。 3. 算子注册验证 - 验证三个算子使用 `OPS_MATH_REGISTER_CPU_KERNELV2` 宏注册成功，算子能正常加载与执行。 4. 二级冒烟 / 算子泛化 - 对 top_k_pq_distance、zeta 执行二级冒烟用例与泛化测试，确认 infershape 迁移后算子行为与预期一致。 ## 文档更新 - 更新 `docs/zh/op_list.md`：top_k_pq_distance、zeta 的 hostcpu infershape 支持状态由 × 更新为 √。 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!4342	1 个月前
op_kernel_aicpu	add infershape for top_k_pq_distance&zeta Co-authored-by: sujunwei3<sujunwei3@huawei.com> # message auto-generated for no-merge-commit merge: !4342 merge dev into master add infershape for top_k_pq_distance&zeta Created-by: sujunwei3 Commit-by: sujunwei3 Merged-by: cann-robot Description: ## 描述为 top_k_pq_distance 与 zeta 算子新增 host 侧 InferShape 实现，并将 top_k_pq_distance、top_k_v2_d、zeta 三个 AICPU 算子的注册宏从 `REGISTER_CPU_KERNEL` 迁移到 `OPS_MATH_REGISTER_CPU_KERNELV2`，同时在对应 CMakeLists.txt 中开启 `HOSTCPU TRUE` 以编译 host 侧 infershape 模块。主要改动： - 新增 `math/top_k_pq_distance/op_host/top_k_pq_distance_infershape.cpp`：三个输出 topk_distance/topk_ivf/topk_index 的 shape 均为 `{k}`（k 来自属性），dtype 分别继承自 pq_distance[0]/pq_ivf[0]/pq_index[0] 的动态输入。 - 新增 `math/zeta/op_host/zeta_infershape.cpp`：输出 z 的 shape 为 x 与 q 的广播结果（复用 `Ops::Base::InferShape4Broadcast`），dtype 继承自 x；并实现 InferShapeRange 以支持动态 shape range 场景。 - 修改 `math/{top_k_pq_distance,top_k_v2_d,zeta}/CMakeLists.txt`：`add_all_modules_sources(... HOSTCPU TRUE)`，启用 hostcpu infershape 编译。 - 修改 `math/{top_k_pq_distance,top_k_v2_d,zeta}/op_kernel_aicpu/_aicpu.cpp`：`REGISTER_CPU_KERNEL` → `OPS_MATH_REGISTER_CPU_KERNELV2`，引入 `aicpu/math_aicpu_register.h`。 - 更新 `docs/zh/op_list.md`：top_k_pq_distance、zeta 的 hostcpu infershape 支持状态由 × 更新为 √。 ## 关联的Issue #2472 https://gitcode.com/cann/ops-math/issues/2472 ## 测试根据代码变更，测试场景如下： 1. 编译构建测试* - 执行整仓编译 `bash build.sh --pkg --jit -j16`，验证三个算子 CMakeLists.txt 的 `HOSTCPU TRUE` 配置变更正确，hostcpu infershape 模块能正常编译产出。 2. InferShape 功能验证 - top_k_pq_distance：验证三个输出 topk_distance/topk_ivf/topk_index 的 shape 均为 `{k}`，dtype 分别与 pq_distance[0]/pq_ivf[0]/pq_index[0] 一致。 - zeta：验证输出 z 的 shape 为 x 与 q 广播后的 shape，dtype 与 x 一致；覆盖静态 shape 与动态 shape range 两种场景（含 0/1/-1 的 range 用例）。 3. 算子注册验证 - 验证三个算子使用 `OPS_MATH_REGISTER_CPU_KERNELV2` 宏注册成功，算子能正常加载与执行。 4. 二级冒烟 / 算子泛化 - 对 top_k_pq_distance、zeta 执行二级冒烟用例与泛化测试，确认 infershape 迁移后算子行为与预期一致。 ## 文档更新 - 更新 `docs/zh/op_list.md`：top_k_pq_distance、zeta 的 hostcpu infershape 支持状态由 × 更新为 √。 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!4342	1 个月前
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	2 个月前
CMakeLists.txt	add infershape for top_k_pq_distance&zeta Co-authored-by: sujunwei3<sujunwei3@huawei.com> # message auto-generated for no-merge-commit merge: !4342 merge dev into master add infershape for top_k_pq_distance&zeta Created-by: sujunwei3 Commit-by: sujunwei3 Merged-by: cann-robot Description: ## 描述为 top_k_pq_distance 与 zeta 算子新增 host 侧 InferShape 实现，并将 top_k_pq_distance、top_k_v2_d、zeta 三个 AICPU 算子的注册宏从 `REGISTER_CPU_KERNEL` 迁移到 `OPS_MATH_REGISTER_CPU_KERNELV2`，同时在对应 CMakeLists.txt 中开启 `HOSTCPU TRUE` 以编译 host 侧 infershape 模块。主要改动： - 新增 `math/top_k_pq_distance/op_host/top_k_pq_distance_infershape.cpp`：三个输出 topk_distance/topk_ivf/topk_index 的 shape 均为 `{k}`（k 来自属性），dtype 分别继承自 pq_distance[0]/pq_ivf[0]/pq_index[0] 的动态输入。 - 新增 `math/zeta/op_host/zeta_infershape.cpp`：输出 z 的 shape 为 x 与 q 的广播结果（复用 `Ops::Base::InferShape4Broadcast`），dtype 继承自 x；并实现 InferShapeRange 以支持动态 shape range 场景。 - 修改 `math/{top_k_pq_distance,top_k_v2_d,zeta}/CMakeLists.txt`：`add_all_modules_sources(... HOSTCPU TRUE)`，启用 hostcpu infershape 编译。 - 修改 `math/{top_k_pq_distance,top_k_v2_d,zeta}/op_kernel_aicpu/_aicpu.cpp`：`REGISTER_CPU_KERNEL` → `OPS_MATH_REGISTER_CPU_KERNELV2`，引入 `aicpu/math_aicpu_register.h`。 - 更新 `docs/zh/op_list.md`：top_k_pq_distance、zeta 的 hostcpu infershape 支持状态由 × 更新为 √。 ## 关联的Issue #2472 https://gitcode.com/cann/ops-math/issues/2472 ## 测试根据代码变更，测试场景如下： 1. 编译构建测试* - 执行整仓编译 `bash build.sh --pkg --jit -j16`，验证三个算子 CMakeLists.txt 的 `HOSTCPU TRUE` 配置变更正确，hostcpu infershape 模块能正常编译产出。 2. InferShape 功能验证 - top_k_pq_distance：验证三个输出 topk_distance/topk_ivf/topk_index 的 shape 均为 `{k}`，dtype 分别与 pq_distance[0]/pq_ivf[0]/pq_index[0] 一致。 - zeta：验证输出 z 的 shape 为 x 与 q 广播后的 shape，dtype 与 x 一致；覆盖静态 shape 与动态 shape range 两种场景（含 0/1/-1 的 range 用例）。 3. 算子注册验证 - 验证三个算子使用 `OPS_MATH_REGISTER_CPU_KERNELV2` 宏注册成功，算子能正常加载与执行。 4. 二级冒烟 / 算子泛化 - 对 top_k_pq_distance、zeta 执行二级冒烟用例与泛化测试，确认 infershape 迁移后算子行为与预期一致。 ## 文档更新 - 更新 `docs/zh/op_list.md`：top_k_pq_distance、zeta 的 hostcpu infershape 支持状态由 × 更新为 √。 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!4342	1 个月前
README.md	md大模型检测低错修改 Co-authored-by: gitcode-chenjiao<chenjiao31@huawei.com> # message auto-generated for no-merge-commit merge: !5183 merge master into master md大模型检测低错修改 Created-by: gitcode-chenjiao Commit-by: gitcode-chenjiao Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> md大模型检测低错修改 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> [# 2466](https://gitcode.com/cann/ops-math/issues/2446) ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ok ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> *.md ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!5183	27 天前

Zeta

产品支持情况

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

功能说明

算子功能：计算Hurwitz zeta函数，即 ζ(x, q) = Σ(n=0 to ∞) 1/(q+n)^x。
计算公式：

\zeta(x, q) = \sum_{n=0}^{\infty} \frac{1}{(q+n)^x}

参数说明

参数名	输入/输出/属性	描述	数据类型	数据格式
x	输入	输入张量，Hurwitz zeta函数的指数参数。	FLOAT、DOUBLE	ND
q	输入	输入张量，Hurwitz zeta函数的偏移参数，必须与x具有相同的数据类型。	FLOAT、DOUBLE	ND
z	输出	输出张量，Hurwitz zeta函数的计算结果，与x具有相同的类型和形状。	FLOAT、DOUBLE	ND

约束说明

输入x和q的数据类型必须一致。
输入x和q的数据大小必须一致（不支持broadcast）。

调用说明

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