master

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

14cc5291Add on 21 days agoHistory

Files	Last commit	Last update
arch35	【mc62】workspaceSize硬编码16M大小，替换为通过接口获取，ascendcPlatform.GetLibApiWorkSpaceSize(); Co-authored-by: xufeng12121<1074805447@qq.com> # message auto-generated for no-merge-commit merge: !5675 merge math_space into master 【mc62】workspaceSize硬编码16M大小，替换为通过接口获取，ascendcPlatform.GetLibApiWorkSpaceSize(); Created-by: xufeng12121 Commit-by: xufeng12121 Merged-by: cann-robot Description: ## 描述 1. 背景：mc62算子workspace占用对比1951增大 2. 由于mc62代码和A5是onetrack, 所有workspacesize硬编码为16M, 在mc62上也同样是16M，asc 提供接口，对mc62进行区分，从原来的16M 变成1K，A5和A2大小不变。 3. 修改：算子侧tiling中硬编码为16M大小的workspacesize统一改为用接口获取。 GetLibApiWorkSpaceSize实现如下： const static uint32_t WORK_SPACE_SIZE_910B = 16 * 1024 * 1024; const static uint32_t WORK_SPACE_SIZE_950 = 16 * 1024 * 1024; const static uint32_t WORK_SPACE_SIZE = 1024; uint32_t PlatformAscendC::GetLibApiWorkSpaceSize(void) const { auto npuArch = GetCurNpuArch(); if (npuArch == NpuArch::DAV_RESV) { PF_LOGE("get platform failed, CurNpuArch is NpuArch::DAV_RESV"); return -1; } else if (npuArch == NpuArch::DAV_2201) { return WORK_SPACE_SIZE_910B; } else if (npuArch == NpuArch::DAV_3510) { return WORK_SPACE_SIZE_950; } return WORK_SPACE_SIZE; } ## 关联的Issue https://gitcode.com/cann/ops-math/issues/3232 ## 测试 1.单算子框架测试通过 2.二级冒烟通过 文档更新 类型标签 - [x] Bug修复 See merge request: cann/ops-math!5675	1 month 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	21 days ago
neg_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	21 days ago
neg_infershape.cpp	支持neg等单目运算算子下一代kernel实现 Co-authored-by: 张磊<zhanglei121@huawei.com> # message auto-generated for no-merge-commit merge: !494 merge master into master 支持neg等单目运算算子下一代kernel实现 Created-by: zl_hw Commit-by: 张磊 Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!494	10 months ago