Star0

Star0

cann-robotfeat: Mod算子适配ascend350(dvlite)平台

b194e795创建于 25 天前历史提交

文件	最后提交记录	最后更新时间
docs	fix(math): release resources on init failure in brc aclnn examples Co-authored-by: xjh2527<522024330101@smail.nju.edu.cn> # message auto-generated for no-merge-commit merge: !4633 merge fix/init-resource-leak-brc-ifaces into master fix(math): release resources on init failure in brc aclnn examples Created-by: xjh2527 Commit-by: xjh2527 Merged-by: cann-robot Description: ## 描述修复 ops-math 仓库中 BRC（二元广播计算）类算子调用示例在初始化失败场景下资源释放不完整的问题。 - `aclrtSetDevice` 失败时，在返回前调用 `aclFinalize`，释放已初始化的 ACL 资源。 - `aclrtCreateStream` 失败时，在返回前依次调用 `aclrtResetDevice` 和 `aclFinalize`，释放已设置的 Device 及 ACL 资源。 - 保留原有 `CHECK_RET` 写法和正常执行路径的资源释放顺序。 - 同步修改文档示例和对应的可执行示例源码，保证两处代码一致。本次修改参考 ops-nn PR #8603（aclnnSoftmax）的修复方式，仅补齐资源释放逻辑，不涉及算子 Tiling、Kernel 和计算功能变更。修复范围严格对应 Issue #2653 中描述的 BRC 算子 ACLNN 接口清单，逐接口匹配对应的 `docs/aclnn.md` 文档与 `examples/test_aclnn_.cpp` 示例。experimental/ 镜像不在本次范围内。 ## 涉及算子共 138 个文件（69 个 `examples/test_aclnn_.cpp` + 69 个 `docs/aclnn.md`），仅覆盖 math/ 和 conversion/ 目录： ### math（120 个文件） - Add：aclnnAdd / aclnnAdds / aclnnInplaceAdd / aclnnInplaceAdds - Sub：aclnnSub / aclnnSubs / aclnnRsub / aclnnRsubs / aclnnInplaceSub / aclnnInplaceSubs - Mul / Muls：aclnnMul / aclnnMuls / aclnnInplaceMul / aclnnInplaceMuls - Minimum / Maximum：aclnnMinimum / aclnnMinN / aclnnMaximum / aclnnMaxN - Signbit：aclnnSignbit - Less / LessEqual：aclnnLtScalar / aclnnLtTensor / aclnnLeScalar / aclnnLeTensor（及 Inplace 变体） - Greater / GreaterEqual：aclnnGtScalar / aclnnGtTensor / aclnnGeScalar / aclnnGeTensor（及 Inplace 变体） - NotEqual / Equal：aclnnNeScalar / aclnnNeTensor / aclnnLogicalXor / aclnnEqScalar / aclnnEqTensor（及 Inplace 变体） - Lerp：aclnnLerp / aclnnLerps / aclnnInplaceLerp / aclnnInplaceLerps - LogicalAnd / LogicalOr：aclnnLogicalAnd / aclnnLogicalOr（及 Inplace 变体） - BitwiseOr / BitwiseAnd / BitwiseXor：aclnnBitwiseOrScalar / aclnnBitwiseOrTensor / aclnnBitwiseAndScalar / aclnnBitwiseAndTensor / aclnnBitwiseXorScalar / aclnnBitwiseXorTensor（及 Inplace 变体） - TanhGrad：aclnnTanhBackward - Addcmul / Ger / Gcd：aclnnAddcmul / aclnnGer / aclnnGcd - SelectV2：aclnnSWhere - RealDiv：aclnnDiv / aclnnDivs / aclnnDivMod / aclnnDivMods（及 Inplace 变体） - FloorDiv：aclnnFloorDivide / aclnnFloorDivides（及 Inplace 变体） - Pow：aclnnExp2 / aclnnPowTensorScalar / aclnnPowTensorTensor / aclnnPowScalarTensor（及 Inplace 变体） - Addcdiv：aclnnAddcdiv / aclnnInplaceAddcdiv - Mod：aclnnFmodScalar / aclnnFmodTensor（及 Inplace 变体） - FloorMod：aclnnRemainderScalarTensor / aclnnRemainderTensorScalar / aclnnRemainderTensorTensor（及 Inplace 变体） ### conversion（18 个文件） - ClipByValueV2：aclnnClamp / aclnnClampMax / aclnnClampMaxTensor / aclnnClampMin / aclnnClampMinTensor / aclnnClampTensor / aclnnHardtanh（及 Inplace 变体）注：upstream 已用 C++ RAII（StreamPtr / aclGuard）重构的 16 个文件本 PR 跳过，不再重复修改。 ## 关联的Issue Issue #2653：https://gitcode.com/cann/ops-math/issues/2653 ## 测试 - `git diff --check`：通过。 - pre-commit 全部通过（clang-format / trailing-whitespace / end-of-file-fixer / codespell / markdownlint / OAT 等）。 - CI 流水线全部通过（cann-cla/yes、api-check-pass、ci-pipeline-passed）：全平台编译、UT 测试、PreSmoke 均成功。 - 本次仅修改调用示例的异常处理逻辑，未执行 NPU 运行测试。 ## 文档更新 - 更新上述算子 `docs/` 目录下的 `aclnn.md` 中的调用示例。 - 同步更新 `examples/` 目录下的 `test_aclnn_.cpp` 可执行示例源码。 ## 类型标签 - [x] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!4633	1 个月前
examples	fix(math): release resources on init failure in brc aclnn examples Co-authored-by: xjh2527<522024330101@smail.nju.edu.cn> # message auto-generated for no-merge-commit merge: !4633 merge fix/init-resource-leak-brc-ifaces into master fix(math): release resources on init failure in brc aclnn examples Created-by: xjh2527 Commit-by: xjh2527 Merged-by: cann-robot Description: ## 描述修复 ops-math 仓库中 BRC（二元广播计算）类算子调用示例在初始化失败场景下资源释放不完整的问题。 - `aclrtSetDevice` 失败时，在返回前调用 `aclFinalize`，释放已初始化的 ACL 资源。 - `aclrtCreateStream` 失败时，在返回前依次调用 `aclrtResetDevice` 和 `aclFinalize`，释放已设置的 Device 及 ACL 资源。 - 保留原有 `CHECK_RET` 写法和正常执行路径的资源释放顺序。 - 同步修改文档示例和对应的可执行示例源码，保证两处代码一致。本次修改参考 ops-nn PR #8603（aclnnSoftmax）的修复方式，仅补齐资源释放逻辑，不涉及算子 Tiling、Kernel 和计算功能变更。修复范围严格对应 Issue #2653 中描述的 BRC 算子 ACLNN 接口清单，逐接口匹配对应的 `docs/aclnn.md` 文档与 `examples/test_aclnn_.cpp` 示例。experimental/ 镜像不在本次范围内。 ## 涉及算子共 138 个文件（69 个 `examples/test_aclnn_.cpp` + 69 个 `docs/aclnn.md`），仅覆盖 math/ 和 conversion/ 目录： ### math（120 个文件） - Add：aclnnAdd / aclnnAdds / aclnnInplaceAdd / aclnnInplaceAdds - Sub：aclnnSub / aclnnSubs / aclnnRsub / aclnnRsubs / aclnnInplaceSub / aclnnInplaceSubs - Mul / Muls：aclnnMul / aclnnMuls / aclnnInplaceMul / aclnnInplaceMuls - Minimum / Maximum：aclnnMinimum / aclnnMinN / aclnnMaximum / aclnnMaxN - Signbit：aclnnSignbit - Less / LessEqual：aclnnLtScalar / aclnnLtTensor / aclnnLeScalar / aclnnLeTensor（及 Inplace 变体） - Greater / GreaterEqual：aclnnGtScalar / aclnnGtTensor / aclnnGeScalar / aclnnGeTensor（及 Inplace 变体） - NotEqual / Equal：aclnnNeScalar / aclnnNeTensor / aclnnLogicalXor / aclnnEqScalar / aclnnEqTensor（及 Inplace 变体） - Lerp：aclnnLerp / aclnnLerps / aclnnInplaceLerp / aclnnInplaceLerps - LogicalAnd / LogicalOr：aclnnLogicalAnd / aclnnLogicalOr（及 Inplace 变体） - BitwiseOr / BitwiseAnd / BitwiseXor：aclnnBitwiseOrScalar / aclnnBitwiseOrTensor / aclnnBitwiseAndScalar / aclnnBitwiseAndTensor / aclnnBitwiseXorScalar / aclnnBitwiseXorTensor（及 Inplace 变体） - TanhGrad：aclnnTanhBackward - Addcmul / Ger / Gcd：aclnnAddcmul / aclnnGer / aclnnGcd - SelectV2：aclnnSWhere - RealDiv：aclnnDiv / aclnnDivs / aclnnDivMod / aclnnDivMods（及 Inplace 变体） - FloorDiv：aclnnFloorDivide / aclnnFloorDivides（及 Inplace 变体） - Pow：aclnnExp2 / aclnnPowTensorScalar / aclnnPowTensorTensor / aclnnPowScalarTensor（及 Inplace 变体） - Addcdiv：aclnnAddcdiv / aclnnInplaceAddcdiv - Mod：aclnnFmodScalar / aclnnFmodTensor（及 Inplace 变体） - FloorMod：aclnnRemainderScalarTensor / aclnnRemainderTensorScalar / aclnnRemainderTensorTensor（及 Inplace 变体） ### conversion（18 个文件） - ClipByValueV2：aclnnClamp / aclnnClampMax / aclnnClampMaxTensor / aclnnClampMin / aclnnClampMinTensor / aclnnClampTensor / aclnnHardtanh（及 Inplace 变体）注：upstream 已用 C++ RAII（StreamPtr / aclGuard）重构的 16 个文件本 PR 跳过，不再重复修改。 ## 关联的Issue Issue #2653：https://gitcode.com/cann/ops-math/issues/2653 ## 测试 - `git diff --check`：通过。 - pre-commit 全部通过（clang-format / trailing-whitespace / end-of-file-fixer / codespell / markdownlint / OAT 等）。 - CI 流水线全部通过（cann-cla/yes、api-check-pass、ci-pipeline-passed）：全平台编译、UT 测试、PreSmoke 均成功。 - 本次仅修改调用示例的异常处理逻辑，未执行 NPU 运行测试。 ## 文档更新 - 更新上述算子 `docs/` 目录下的 `aclnn.md` 中的调用示例。 - 同步更新 `examples/` 目录下的 `test_aclnn_.cpp` 可执行示例源码。 ## 类型标签 - [x] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!4633	1 个月前
framework	fix(mod,mul,pow): 修复日志拼写错误（surpport/UnSupported/faild/intergal/form→from 等） Co-authored-by: sun-yibo1<sunyibo1@hisilicon.com> # message auto-generated for no-merge-commit merge: !4949 merge fix/mod-mul-pow-log-typo into master fix(mod,mul,pow): 修复日志拼写错误（surpport/UnSupported/faild/intergal/form→from 等） Created-by: sun-yibo1 Commit-by: sun-yibo1 Merged-by: cann-robot Description: ## 描述参照 `log_audit_report.md`，修复 `mod`、`mul`、`pow` 三个算子的日志拼写错误（共 11 处）。所有改动仅涉及日志文案，不改动任何逻辑。 ### 改动原因这些算子的实现文件中存在拼写错误，会误导日志排查。 ### 改动方法仅修改日志文案中的拼写错误，不改动任何逻辑代码。同时按 pre-commit `clang-format` 要求对文件做格式化（部分文件在 upstream 上即不满足 `.clang-format`，hook 强制格式化才能通过；纯格式，无语义变更）。 ### 涉及修复 mod（3 处）： - `mod_onnx_plugin.cpp`：`surpport` → `support` - `mod_tiling_arch35.cpp`：`form` → `from`（2 处，`Get ubSize form compileInfo/ascendcPlatform`） mul（7 处）： - `mul.cpp`：`UnSupported` → `Unsupported`（2 处）；`do no support inplace` → `do not support inplace`（2 处） - `mul_tiling_arch35.cpp`：`form` → `from`（2 处） - `aclnn_mul.cpp`：`only support` → `only supports` pow（2 处）： - `pow_tensor_tensor_tiling_arch35.cpp`：`faild` → `failed` - `aclnn_pow.cpp`：`intergal` → `integral` ## 关联的Issue - #2629 ## 测试 - 仅日志文案/格式修改，无逻辑变更，不影响编译与功能。 - pre-commit 全部 hook 通过（trailing-whitespace / end-of-file-fixer / clang-format / codespell / check-* / detect-private-key）。 ## 文档更新无。 ## 类型标签 - [x] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： ## AI/Agent生成声明 - [x] AI辅助编写 See merge request: cann/ops-math!4949	1 个月前
op_api	Mod supports Int64 Co-authored-by: HelloWord-LuXin<luxin67@h-partners.com> # message auto-generated for no-merge-commit merge: !1122 merge modInt64 into master Mod supports Int64 Created-by: HelloWord-LuXin Commit-by: HelloWord-LuXin Merged-by: cann-robot Description: ## 描述 mod算子支持int64 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> ## 关联的Issue https://gitcode.com/cann/ops-math/issues/753 <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> 冒烟，单算子，aclnn调用，pta调用，泛化测试，极值边界值等自测。 ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> aclnnFmodTensor&aclnnInplaceFmodTensor.md ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!1122	7 个月前
op_graph	range等算子资料更新 Co-authored-by: suxiaolan2000<jiangyulong10@huawei.com> # message auto-generated for no-merge-commit merge: !697 merge math同步 into master range等算子资料更新 Created-by: suxiaolan2000 Commit-by: suxiaolan2000 Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!697	8 个月前
op_host	feat: Mod算子适配ascend350(dvlite)平台 Co-authored-by: esok11<yangsifa@huawei.com> # message auto-generated for no-merge-commit merge: !5540 merge f0910 into master feat: Mod算子适配ascend350(dvlite)平台 Created-by: esok11 Commit-by: esok11 Merged-by: cann-robot Description: ## 描述为 Mod 算子适配 ascend350（dvlite）平台，复用现有 arch35 tiling 实现，使 Mod 算子可在 ascend350 环境下编译与执行。改动原因：Mod 算子此前仅支持 ascend950、mc62 平台，需要扩展对 ascend350（dvlite）平台的支持。改动方法： 1. `math/mod/CMakeLists.txt`：`SUPPORT_COMPUTE_UNIT` 新增 `ascend350`，`SUPPORT_TILING_DIR` 对应新增 `arch35`（与 ascend950/mc62 共用同一 tiling 目录）。 2. `math/mod/op_host/mod_def.cpp`：新增 `this->AICore().AddConfig("ascend350", aicoreConfig)`，为 ascend350 注册 AICore 算子配置（DynamicCompileStaticFlag(true)、DynamicFormatFlag(false)、DynamicRankSupportFlag(true)、DynamicShapeSupportFlag(true)、NeedCheckSupportFlag(false)、PrecisionReduceFlag(true)、ExtendCfgInfo("opFile.value", "mod_apt")），与 ascend950 配置一致；其余为代码格式化调整，无功能变更。 3. `math/mod/op_host/config/ascend350/mod_binary.json`（新增）：ascend350 平台的二进制算子配置，覆盖 bfloat16/float16/float32/int32/uint8/int8/int64 共 7 种数据类型（输入 x1、x2 与输出 y 均为 ND 格式、动态 shape、FormatAgnostic），内容与 ascend950 的 mod_binary.json 一致。 ## 关联的Issue - #3119 ## 测试本 PR 未包含测试用例改动，待在 ascend350 环境完成二级冒烟、算子泛化等验证。 ## 文档更新本 PR 不涉及文档更新。 ## 类型标签 - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!5540	25 天前
op_kernel	refactor: arch35算子Reg级API重命名迁移 Co-authored-by: hahaha22<wangrunze20@h-partners.com> # message auto-generated for no-merge-commit merge: !4468 merge work_8 into master refactor: arch35算子Reg级API重命名迁移 Created-by: hahaha22 Commit-by: hahaha22 Merged-by: cann-robot Description: ## 描述将 arch35 架构下 40 个算子 kernel 文件中的 Reg 级 API 名称从旧命名迁移到新命名，与已迁移的算子（xlogy、dawsn、rint、cdist、reduce_all、power 等）保持一致。涉及的 API 重命名映射： - `__local_mem__` → `__ubuf__`（地址空间限定符统一） - `Reg::DataCopy` (LoadDist) → `Reg::LoadAlign`（UB 到寄存器的对齐加载） - `Reg::DataCopy` (StoreDist) → `Reg::StoreAlign`（寄存器到 UB 的对齐存储） - `Reg::DataCopyUnAlignPre` → `Reg::LoadUnAlignPre` - `Reg::DataCopyUnAlign` → `Reg::LoadUnAlign` - `Reg::MaskPack` → `Reg::Pack` - `Reg::MaskUnPack` → `Reg::UnPack` - `Reg::MaskAnd` → `Reg::And`（与已有 Reg::And 通过重载统一） - `Reg::CompareScalar` → `Reg::Compares` - `Reg::UnalignReg` → `Reg::UnalignRegForLoad` 涉及的算子包括：clip_by_value、clip_by_value_v2、adjacent_difference、bitwise_xor、div、floor_div、floor_mod、lerp、logical_or、mod、pow、signbit、tanh_grad。此外，adjacent_difference 算子中存在原有的 `__local_mem__`/`__ubuf__` 地址空间限定符不一致问题（左侧声明与右侧转换不匹配），本次重命名一并修复。 ### 改动原因 AscendC SDK 更新了 Reg 级 API 命名规范，将原有的 `DataCopy` 拆分为语义更明确的 `LoadAlign`/`StoreAlign`，并移除了 Mask 系列 API 的 "Mask" 前缀（通过重载统一）。本 PR 将相关算子迁移至新命名。 ### 改动方法纯机械重命名，无逻辑变更。所有改动均为 1:1 的 API 名称替换，增删行数完全对称（+417 -417）。Load/Store 语义区分基于原有模板参数（LoadDist → LoadAlign, StoreDist → StoreAlign）和参数顺序（寄存器在前为 Load，地址在前为 Store）。 ## 关联的Issue - #2529 ## 测试待补充：二级冒烟测试、算子泛化测试。 ## 文档更新无 ## 类型标签 - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [x] 其他，请描述：API 重命名重构 See merge request: cann/ops-math!4468	1 个月前
tests	add ttk_kernel st csv cases Co-authored-by: git config --global user.name gcw_CRjbEjwl<fengdaoyong@h-partners.com> # message auto-generated for no-merge-commit merge: !3379 merge 20260616 into master add ttk_kernel st csv cases Created-by: gcw_CRjbEjwl Commit-by: git config --global user.name gcw_CRjbEjwl Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> 添加算子仓内arch35下的ttk用例 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> 不涉及 ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> 本地验证 ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [x] 其他，请描述： See merge request: cann/ops-math!3379	3 个月前
CMakeLists.txt	feat: Mod算子适配ascend350(dvlite)平台 Co-authored-by: esok11<yangsifa@huawei.com> # message auto-generated for no-merge-commit merge: !5540 merge f0910 into master feat: Mod算子适配ascend350(dvlite)平台 Created-by: esok11 Commit-by: esok11 Merged-by: cann-robot Description: ## 描述为 Mod 算子适配 ascend350（dvlite）平台，复用现有 arch35 tiling 实现，使 Mod 算子可在 ascend350 环境下编译与执行。改动原因：Mod 算子此前仅支持 ascend950、mc62 平台，需要扩展对 ascend350（dvlite）平台的支持。改动方法： 1. `math/mod/CMakeLists.txt`：`SUPPORT_COMPUTE_UNIT` 新增 `ascend350`，`SUPPORT_TILING_DIR` 对应新增 `arch35`（与 ascend950/mc62 共用同一 tiling 目录）。 2. `math/mod/op_host/mod_def.cpp`：新增 `this->AICore().AddConfig("ascend350", aicoreConfig)`，为 ascend350 注册 AICore 算子配置（DynamicCompileStaticFlag(true)、DynamicFormatFlag(false)、DynamicRankSupportFlag(true)、DynamicShapeSupportFlag(true)、NeedCheckSupportFlag(false)、PrecisionReduceFlag(true)、ExtendCfgInfo("opFile.value", "mod_apt")），与 ascend950 配置一致；其余为代码格式化调整，无功能变更。 3. `math/mod/op_host/config/ascend350/mod_binary.json`（新增）：ascend350 平台的二进制算子配置，覆盖 bfloat16/float16/float32/int32/uint8/int8/int64 共 7 种数据类型（输入 x1、x2 与输出 y 均为 ND 格式、动态 shape、FormatAgnostic），内容与 ascend950 的 mod_binary.json 一致。 ## 关联的Issue - #3119 ## 测试本 PR 未包含测试用例改动，待在 ascend350 环境完成二级冒烟、算子泛化等验证。 ## 文档更新本 PR 不涉及文档更新。 ## 类型标签 - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!5540	25 天前
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	1 个月前

Mod

产品支持情况

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

功能说明

算子功能：返回self除以other的余数。
计算公式：

将张量self，和标量或张量other，进行广播成相同shape的张量后，Mod可以用如下数学公式表示：
$outi=selfi−trunc(selfiotheri)∗otheriout_{i} = self_{i} - trunc \left(\frac{self_{i}}{other_{i}}\right) * other_{i}$

参数说明

参数名	输入/输出/属性	描述	数据类型	数据格式
self	输入	待进行mod计算的入参，公式中的self_i。	DOUBLE、BFLOAT16、FLOAT16、FLOAT32、INT32、INT64、INT8、UINT8	ND
other	输入	待进行mod计算的入参，公式中的other_i。	DOUBLE、BFLOAT16、FLOAT16、FLOAT32、INT32、INT64、INT8、UINT8	ND
out	输出	待进行mod计算的出参，公式中的out_i。	DOUBLE、BFLOAT16、FLOAT16、FLOAT32、INT32、INT64、INT8、UINT8	ND

约束说明

数据类型需满足数据类型推导规则，推导后的数据类型需在支持的数据类型范围内。
self和other的shape必须满足广播规则。
数据维度不支持8维以上。

调用说明

调用方式	样例代码	说明
aclnn接口	test_aclnn_fmod_scalar	通过aclnnFmodScalar接口方式调用Mod算子。
aclnn接口	test_aclnn_fmod_tensor	通过aclnnFmodTensor接口方式调用Mod算子。