Star0

Star0

cann-robottest: 18 算子 golden 新增 TestSpec 双注册与 third_party

aa36dc1d创建于 20 天前历史提交

文件	最后提交记录	最后更新时间
examples	format: batch_to_space_nd 等算子代码格式化 Co-authored-by: zhanw_coding<zhanwang@huawei.com> # message auto-generated for no-merge-commit merge: !3822 merge fix-fmt into master format: batch_to_space_nd 等算子代码格式化 Created-by: zhanw_coding Commit-by: zhanw_coding Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> format: batch_to_space_nd 等算子代码格式化 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> https://gitcode.com/cann/ops-math/issues/2147 ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [x] 其他，请描述：代码格式化 See merge request: cann/ops-math!3822	3 个月前
framework	迁移开源TF插件--math Co-authored-by: yaochen<yaochen15@huawei.com> # message auto-generated for no-merge-commit merge: !4308 merge transfer-math into master 迁移开源TF插件--math Created-by: nextyale Commit-by: yaochen Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [x] 其他，请描述：迁移TF插件 See merge request: cann/ops-math!4308	2 个月前
op_graph	math仓产品名称修改 Co-authored-by: sunchun<sunchun4@h-partners.com> # message auto-generated for no-merge-commit merge: !5986 merge atlas into master math仓产品名称修改 Created-by: sunchun Commit-by: sunchun Merged-by: cann-robot Description: ## 描述统一 math 仓 5 个算子 proto.h 注释中的产品名称描述，使其符合最新产品命名规范： - conversion/dynamic_stitch：`Atlas A2 Training Series Product/...` → `Atlas A2 products and Atlas A3 training products` - experimental/math/gcd：`Atlas A2/A3 Ascend C path` → `Atlas A2 products and Atlas A3 products Ascend C path` - math/cast：`Atlas Inference Series Product` → `Atlas inference products` - math/truncate_mod：`Atlas 200/300/500 Inference Product` → `Ascend 950PR&950DT products` - random/drop_out_do_mask：统一为 `Atlas A2 products` / `Atlas A3 training/inference products` ## 关联的Issue - #3314 https://gitcode.com/cann/ops-math/issues/3314 ## 测试仅修改文档注释，不涉及代码逻辑变更，无需进行功能测试。 ## 文档更新更新了上述 5 个算子 proto.h 中的产品名称注释。 ## 类型标签 - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!5986	1 个月前
op_host	feat: 新增 ascend960pr/ascend960dt 平台算子适配（14 个算子） Co-authored-by: esok11<yangsifa@huawei.com> # message auto-generated for no-merge-commit merge: !6157 merge f1008 into master feat: 新增 ascend960pr/ascend960dt 平台算子适配（14 个算子） Created-by: esok11 Commit-by: esok11 Merged-by: cann-robot Description: ## 描述为 ascend960pr / ascend960dt 两个新增芯片形态适配算子，覆盖 conversion 与 math 目录下共 14 个算子；同时修正 addr 的 dtype 支持列表判断逻辑，并对个别文件做版权头/格式整理。 ### 改动原因 ascend960pr / ascend960dt 为新增芯片形态，现有算子需补齐平台注册、编译单元与二进制配置后，才能在新平台上被算子信息库发现、编译并选择正确的 kernel 二进制。 ### 改动方法 1. 算子信息库注册（`_def.cpp`）：新增 `this->AICore().AddConfig("ascend960pr", ...)` 与 `AddConfig("ascend960dt", ...)` - conversion：assign、dynamic_partition、masked_fill、mem_set、tensor_move、tril、triu - math：addr、asin、eye、mod、sinc、truncate_div、truncate_mod 2. 编译配置（`CMakeLists.txt`）：在 `SUPPORT_COMPUTE_UNIT` / `COMPUTE_UNIT` 中追加 `ascend960pr`、`ascend960dt`，并在 `SUPPORT_TILING_DIR` / `TILING_DIR` 中追加对应的 `arch35`（tiling 目录沿用 arch35） 3. 二进制配置：为每个算子新增 `op_host/config/ascend960pr/` 与 `op_host/config/ascend960dt/` 目录，包含 `<op>_binary.json` 与 `<op>_simplified_key.ini` 4. addr dtype 判断修正（`math/addr/op_api/aclnn_addr.cpp`）：`GetDtypeSupportList()` 的判断由区间判断 `>= ASCEND910B && <= ASCEND910E` 改为显式判断 `== ASCEND910B \|\| == ASCEND910_93 \|\| IsRegBase()`，使 reg-base 平台复用 910B dtype 支持列表 5. 格式整理*：`conversion/mem_set_v2/op_host/mem_set_v2_def.cpp`、`math/adds/op_host/adds_def.cpp`、`math/adds/CMakeLists.txt` 仅做版权头/空行/行尾换行等整理 ## 关联的Issue - https://gitcode.com/cann/ops-math/issues/3440 ## 测试本 PR 未包含测试代码改动；适配涉及算子信息库、编译配置与二进制配置，需在 ascend960pr / ascend960dt 平台补充二级冒烟与算子泛化验证。 ## 文档更新无。 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!6157	23 天前
op_kernel	format: batch_to_space_nd 等算子代码格式化 Co-authored-by: zhanw_coding<zhanwang@huawei.com> # message auto-generated for no-merge-commit merge: !3822 merge fix-fmt into master format: batch_to_space_nd 等算子代码格式化 Created-by: zhanw_coding Commit-by: zhanw_coding Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> format: batch_to_space_nd 等算子代码格式化 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> https://gitcode.com/cann/ops-math/issues/2147 ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [x] 其他，请描述：代码格式化 See merge request: cann/ops-math!3822	3 个月前
tests	test: 18 算子 golden 新增 TestSpec 双注册与 third_party Co-authored-by: Xbying<2119046648@qq.com> # message auto-generated for no-merge-commit merge: !5789 merge 14 into master test: 18 算子 golden 新增 TestSpec 双注册与 third_party Created-by: 2301_81474000 Commit-by: Xbying Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> - 16 个算子 golden 新增 __spec__（保留旧 __golden__，单一代码源双入口）：Pad、PadV2、PadV3、MirrorPad、CircularPadGrad、PadV3GradReplication、DepthToSpace、SpaceToDepth、BatchToSpaceND、MatrixSetDiagV2、Sinc、TruncateDiv、TruncateMod、Erfc、Expand、Tile/TileD - 新建 2 个 golden：ReflectionPad3dGrad、Rint - 全部补 third_party 竞品实现（F.pad / aten backward / np.pad / torch.tile / torch.round 等），视图类输出加 .contiguous() - 顺带修复 4 处原 golden 缺陷（depth_to_space 死链与未定义名、pad_v2 0 维标量越界、pad_v3_grad_replication 部分维 pad、ruff E712/F841） - 全部经 TTK loader 双入口解析与 third_party↔golden 数值互验（BatchToSpaceND 因 golden 依赖 TF 仅做解析验证） ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> https://gitcode.com/cann/ops-math/issues/3226 ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [x] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!5789	20 天前
CMakeLists.txt	feat: 新增 ascend960pr/ascend960dt 平台算子适配（14 个算子） Co-authored-by: esok11<yangsifa@huawei.com> # message auto-generated for no-merge-commit merge: !6157 merge f1008 into master feat: 新增 ascend960pr/ascend960dt 平台算子适配（14 个算子） Created-by: esok11 Commit-by: esok11 Merged-by: cann-robot Description: ## 描述为 ascend960pr / ascend960dt 两个新增芯片形态适配算子，覆盖 conversion 与 math 目录下共 14 个算子；同时修正 addr 的 dtype 支持列表判断逻辑，并对个别文件做版权头/格式整理。 ### 改动原因 ascend960pr / ascend960dt 为新增芯片形态，现有算子需补齐平台注册、编译单元与二进制配置后，才能在新平台上被算子信息库发现、编译并选择正确的 kernel 二进制。 ### 改动方法 1. 算子信息库注册（`_def.cpp`）：新增 `this->AICore().AddConfig("ascend960pr", ...)` 与 `AddConfig("ascend960dt", ...)` - conversion：assign、dynamic_partition、masked_fill、mem_set、tensor_move、tril、triu - math：addr、asin、eye、mod、sinc、truncate_div、truncate_mod 2. 编译配置（`CMakeLists.txt`）：在 `SUPPORT_COMPUTE_UNIT` / `COMPUTE_UNIT` 中追加 `ascend960pr`、`ascend960dt`，并在 `SUPPORT_TILING_DIR` / `TILING_DIR` 中追加对应的 `arch35`（tiling 目录沿用 arch35） 3. 二进制配置：为每个算子新增 `op_host/config/ascend960pr/` 与 `op_host/config/ascend960dt/` 目录，包含 `<op>_binary.json` 与 `<op>_simplified_key.ini` 4. addr dtype 判断修正（`math/addr/op_api/aclnn_addr.cpp`）：`GetDtypeSupportList()` 的判断由区间判断 `>= ASCEND910B && <= ASCEND910E` 改为显式判断 `== ASCEND910B \|\| == ASCEND910_93 \|\| IsRegBase()`，使 reg-base 平台复用 910B dtype 支持列表 5. 格式整理*：`conversion/mem_set_v2/op_host/mem_set_v2_def.cpp`、`math/adds/op_host/adds_def.cpp`、`math/adds/CMakeLists.txt` 仅做版权头/空行/行尾换行等整理 ## 关联的Issue - https://gitcode.com/cann/ops-math/issues/3440 ## 测试本 PR 未包含测试代码改动；适配涉及算子信息库、编译配置与二进制配置，需在 ascend960pr / ascend960dt 平台补充二级冒烟与算子泛化验证。 ## 文档更新无。 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!6157	23 天前
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 个月前

TruncateMod

产品支持情况

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

功能说明

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

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

参数说明

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

约束说明

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

调用说明

调用方式	样例代码	说明
geir接口	test_geir_truncate_mod	通过geir接口方式调用TruncateMod算子。