| Change blockDim to numBlocks
Co-authored-by: chupeng1996<chupeng5@huawei.com>
# message auto-generated for no-merge-commit merge:
!235 merge fix_num into master
Change blockDim to numBlocks
Created-by: chupeng1996
Commit-by: chupeng1996
Merged-by: cann-robot
Description: ## 描述
<!--在这里详细描述你的改动,包括改动的原因和所采取的方法。-->
问题/功能描述
本次PR旨在对AscendC内核启动、元数据生成及后处理相关的代码进行一系列重构和修复,以提升代码的可读性、一致性和准确性。核心内容包括:统一将表示内核执行块数量的变量、参数及结构体名称从blockDim重命名为numBlocks,使其更准确地反映其“块数量”的语义;修复了元数据生成代码中因结构体名称错误(FuncMetaBlockDim)导致的潜在编译或链接问题。
修改方案描述
修改涉及多个文件,主要包括:1)在Python脚本(extract_host_stub.py)中,将混合向量核心启动逻辑及相关函数(如LaunchAscendKernel)中的变量名(如__ascendc_aicBlockDim)统一更新为__ascendc_aicNumBlocks;2)在C++头文件中,将内核元数据结构体FuncMetaBlockDim重命名为FuncMetaNumBlocks;3)在Python元数据生成器(DFXSectionGenerator)中,将生成块数量元信息节的结构体名称和变量名更正为FuncMetaNumBlocks与_kernel_metainfo_numblocks_section;4)在TVM后处理函数(tvm_callback_cce_postproc)中,将参数blockdim和subblockdim分别重命名为numBlocks和subNumBlocks。所有修改均为命名优化和符号修正,不改变功能逻辑。
## 关联的Issue
<!-- 如果这个PR是为了解决特定的Issue,请在这里提供Issue链接。例如:关联Issue #123-->
<!-- 如果这个PR是为了解决特定的问题单,请在这里描述问题单单号。-->
NA
## 测试
<!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。-->
编包装包正常,自定义算子工程用例跑通。
## 文档更新
<!--如果这个PR包含文档的更新,请在这里指出。例如:更新了README.md文件。-->
NA
## 类型标签
<!-- [x] 表示选中 -->
- [x] Bug修复
- [ ] 新特性
- [ ] 性能优化
- [ ] 文档更新
- [ ] 其他,请描述:
See merge request: cann/asc-devkit!235 | 3 个月前 |