master

cann-robotIsClose等15个算子新增支持 ascend960dt

14cc5291Add on 24 days agoHistory

Files	Last commit	Last update
arch35	批量修改ws=0 Co-authored-by: ligen<ligen75@h-partners.com> # message auto-generated for no-merge-commit merge: !6212 merge master into master 批量修改ws=0 Created-by: ligen75 Commit-by: ligen Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> 批量修改算子workspace=0 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> https://gitcode.com/cann/ops-math/issues/3013 ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> 自验通过 ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!6212	30 days ago
config	IsClose等15个算子新增支持 ascend960dt Co-authored-by: TH<taohai2@huawei.com> # message auto-generated for no-merge-commit merge: !6079 merge backBlue into master IsClose等15个算子新增支持 ascend960dt Created-by: TH_HW Commit-by: TH Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> 本次改动为 math 仓 15 个算子新增 Ascend960DT（NpuArch::DAV_9201）平台支持，使这些算子可在 Ascend960DT 上完成注册、编译与运行。 适配算子：is_close、is_neg_inf、masked_scale、neg、reciprocal、round、rsqrt、rsqrt_grad、sign、sin、sqrt、sqrt_grad、square、tanh、trunc。 具体改动： - 算子注册：在各算子 op_host/*_def.cpp 中新增 this->AICore().AddConfig("ascend960dt", aicoreConfig)，完成 Ascend960DT 计算单元注册。 - 编译单元：在各算子 CMakeLists.txt 的 SUPPORT_COMPUTE_UNIT 中加入 ascend960dt，并按原有平台（arch22/arch35）在 SUPPORT_TILING_DIR 中补入对应的 arch35 tiling 目录。 - 二进制配置：新增 op_host/config/ascend960dt/<op>_binary.json，覆盖各算子的 dtype/shape 组合（如 float16/bfloat16/float32/int32/int8/uint8 等）；新增 <op>_simplified_key.ini 指定 --simplified_key_mode。 - 内核编译配置：在 scripts/kernel/binary_config/ascendc_config.json 中为上述算子的 compute_units 与 compile_options 增加 ascend960dt。 - 平台映射：在 scripts/kernel/binary_script/build_env.sh 的 SOC_MAP 中新增 [Ascend960DT]="Ascend960DT_968B7"。 - 算子校验：math/sign/op_api/aclnn_sign.cpp 的 dtype 校验新增 NpuArch::DAV_9201 分支，使 bf16 支持在 Ascend960DT 生效。 - 用例：补充 sin、sqrt 的 arch35 tiling UT（新增小 shape、大 shape、3D、非支持 dtype 等正向/异常用例）；sqrt 用例同时做格式化整改。 - 顺带统一部分 CMakeLists.txt 的许可头注释格式（add_n、complex_abs、cast、is_inf、log1p、tanh 等），并修正 neg 的 ascend950 simplified_key.ini 末尾换行。 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> NA ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> - 编译验证：执行 math 仓编包，确认 15 个算子在 Ascend960DT 下均能正常注册与编译，生成对应 kernel 二进制。 - 二进制/交付件验证：确认各算子 config/ascend960dt 下的 *_binary.json 与 *_simplified_key.ini 生效，ascendc_config.json 中已包含 ascend960dt 编译单元。 - 单元测试：执行 sin、sqrt 的 arch35 tiling UT，覆盖 fp16/bf16/fp32、小/大 shape、3D 及非支持 dtype 场景，结果符合预期。 ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> 无 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!6079	24 days ago
rsqrt_def.cpp	IsClose等15个算子新增支持 ascend960dt Co-authored-by: TH<taohai2@huawei.com> # message auto-generated for no-merge-commit merge: !6079 merge backBlue into master IsClose等15个算子新增支持 ascend960dt Created-by: TH_HW Commit-by: TH Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> 本次改动为 math 仓 15 个算子新增 Ascend960DT（NpuArch::DAV_9201）平台支持，使这些算子可在 Ascend960DT 上完成注册、编译与运行。 适配算子：is_close、is_neg_inf、masked_scale、neg、reciprocal、round、rsqrt、rsqrt_grad、sign、sin、sqrt、sqrt_grad、square、tanh、trunc。 具体改动： - 算子注册：在各算子 op_host/*_def.cpp 中新增 this->AICore().AddConfig("ascend960dt", aicoreConfig)，完成 Ascend960DT 计算单元注册。 - 编译单元：在各算子 CMakeLists.txt 的 SUPPORT_COMPUTE_UNIT 中加入 ascend960dt，并按原有平台（arch22/arch35）在 SUPPORT_TILING_DIR 中补入对应的 arch35 tiling 目录。 - 二进制配置：新增 op_host/config/ascend960dt/<op>_binary.json，覆盖各算子的 dtype/shape 组合（如 float16/bfloat16/float32/int32/int8/uint8 等）；新增 <op>_simplified_key.ini 指定 --simplified_key_mode。 - 内核编译配置：在 scripts/kernel/binary_config/ascendc_config.json 中为上述算子的 compute_units 与 compile_options 增加 ascend960dt。 - 平台映射：在 scripts/kernel/binary_script/build_env.sh 的 SOC_MAP 中新增 [Ascend960DT]="Ascend960DT_968B7"。 - 算子校验：math/sign/op_api/aclnn_sign.cpp 的 dtype 校验新增 NpuArch::DAV_9201 分支，使 bf16 支持在 Ascend960DT 生效。 - 用例：补充 sin、sqrt 的 arch35 tiling UT（新增小 shape、大 shape、3D、非支持 dtype 等正向/异常用例）；sqrt 用例同时做格式化整改。 - 顺带统一部分 CMakeLists.txt 的许可头注释格式（add_n、complex_abs、cast、is_inf、log1p、tanh 等），并修正 neg 的 ascend950 simplified_key.ini 末尾换行。 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> NA ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> - 编译验证：执行 math 仓编包，确认 15 个算子在 Ascend960DT 下均能正常注册与编译，生成对应 kernel 二进制。 - 二进制/交付件验证：确认各算子 config/ascend960dt 下的 *_binary.json 与 *_simplified_key.ini 生效，ascendc_config.json 中已包含 ascend960dt 编译单元。 - 单元测试：执行 sin、sqrt 的 arch35 tiling UT，覆盖 fp16/bf16/fp32、小/大 shape、3D 及非支持 dtype 场景，结果符合预期。 ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> 无 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!6079	24 days ago
rsqrt_infershape.cpp	round,rsqrt,rsqrt_grad,sort算子支持下一代实现 Co-authored-by: Huang-Peng<huangpeng98@huawei.com> # message auto-generated for no-merge-commit merge: !557 merge trans-part1228 into master round,rsqrt,rsqrt_grad,sort算子支持下一代实现 Created-by: Huang-Peng Commit-by: Huang-Peng Merged-by: cann-robot Description: ## 描述 round,rsqrt,rsqrt_grad,sort算子支持最新实现 ## 关联的Issue 不涉及 ## 测试 已完成本地编译、UT、二级冒烟等测试 ## 文档更新 更新了round,rsqrt,rsqrt_grad,sort等算子的README实现 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!557	10 months ago