Files	Last commit	Last update
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	3 months ago
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	2 months ago
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	3 months ago
op_kernel_aicpu	fix(aicpu): 整改代码质量告警 Co-authored-by: zhaowenrui666<zhaowenrui7@huawei.com> # message auto-generated for no-merge-commit merge: !5881 merge fix-aicpu-codecheck-warnings into master fix(aicpu): 整改代码质量告警 Created-by: zhaowenrui666 Commit-by: zhaowenrui666 Merged-by: cann-robot Description: ## 描述整改部分 AICPU 算子的代码质量告警： - 将可替代的 `reinterpret_cast` 调整为 `static_cast`。 - 为 lambda 显式列出捕获变量，避免默认引用捕获。 - 补充成员函数 `const` 限定、析构函数 `override` 以及头文件内函数的 `inline` 限定。 - 修正有符号整数左移、状态码隐式布尔转换及指针顶层 `const` 等告警。 - 同步头文件声明与实现签名，不改变算子功能语义。 ## 关联的Issue https://gitcode.com/cann/ops-math/issues/3264 ## 测试 - `cmake --build build --target aicpu_kernels -j8`：通过，成功生成 `libmath_aicpu_kernels.so`。 - 对本次修改文件执行 `pre-commit run --files ...`：全部通过，包括 clang-format、codespell 和 OAT 检查。 - `git diff --check`：通过。 ## 文档更新无文档变更。 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [x] 其他，请描述：AICPU 代码质量告警整改 See merge request: cann/ops-math!5881	1 month ago
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	3 months ago
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	3 months ago
README.md	产品名变更 Co-authored-by: gitcode-chenjiao<chenjiao31@huawei.com> # message auto-generated for no-merge-commit merge: !5912 merge master into master 产品名变更 Created-by: gitcode-chenjiao Commit-by: gitcode-chenjiao Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> AI处理器对应的产品型号名称更新 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> [#3258](https://gitcode.com/cann/ops-math/issues/3258) ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ok ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> *.md ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!5912	1 month ago
README_en.md	docs: add English translations for markdown files Co-authored-by: jenny_xiang<jenny.xiang@huawei.com> # message auto-generated for no-merge-commit merge: !7113 merge docs/add-english-translations into master docs: add English translations for markdown files Created-by: jenny_xiang Commit-by: jenny_xiang Merged-by: cann-robot Description: ## 描述新增仓库Markdown文档英文译稿（第二批，共409个文件）：第一批（已合入）之外的其余全部译稿，覆盖conversion、examples、ge、math、random各类算子README译稿，以及CHANGELOG_en.md、.gitcode PR模板英文版等。本批内全部仓内链接均指向master已有文件（含第一批已合入的21+2篇译稿），合入后本仓432篇英文文档全部就位，链接完整生效。本批同时包含4篇基于最新源文档的增量更新（math/ada_cast、math/prod_force_se_a、math/sinkhorn、math/xlogy，对应上游近期文档变更）。 ## 关联的Issue 无 ## 测试文档类变更，无需算子测试；已通过链接有效性校验（本批全部相对链接仅指向master已有文件，无指向本PR外未提交文件的链接），并完成逐篇翻译规范自查与语义校对（含支持状态表、限定词、术语、数值符号核对）。 ## 文档更新新增409个英文译稿文件（*_en.md / .en-US.md），不修改任何现有文件。 ## 类型标签 - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!7113	9 days ago

Zeta

English | 简体中文

Product Support Status

Product	Supported
Ascend 950PR&950DT products	√
Atlas A3 products	√
Atlas A2 products	√
Atlas 200I/500 A2 inference products	√
Atlas inference products	√
Atlas training products	√

Function Description

Operator function: computes the Hurwitz zeta function, that is, ζ(x, q) = Σ(n=0 to ∞) 1/(q+n)^x.
Computation formula:

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

Parameter Description

Parameter Name	Input/Output/Attribute	Description	Data Type	Data Format
x	Input	Input tensor, which is the exponent parameter of the Hurwitz zeta function.	FLOAT, DOUBLE	ND
q	Input	Input tensor, which is the shift parameter of the Hurwitz zeta function. It must have the same data type as x.	FLOAT, DOUBLE	ND
z	Output	Output tensor, which is the computation result of the Hurwitz zeta function and has the same type and shape as x.	FLOAT, DOUBLE	ND

Constraint Description

The inputs x and q must have the same data type.
The inputs x and q must have the same data size (broadcast is not supported).

Invocation Description

Invocation Method	Sample Code	Description
Graph-mode invocation	test_geir_zeta	Invokes the Zeta operator through the operator IR API.