
GetResGroupBarrierWorkSpaceSize【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言原生支持C和C标准规范主要由类库和语言扩展层构成提供多层级API满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit功能说明获取GroupBarrier所需要的workspace空间大小。函数原型uint32_t GetResGroupBarrierWorkSpaceSize(void) const参数说明无返回值说明当前GroupBarrier所需要的workspace大小。约束说明无。调用示例// 用户自定义的tiling函数 static ge::graphStatus TilingFunc(gert::TilingContext* context) { AddApiTiling tiling; ... // 如需要使用系统workspace需要调用GetLibApiWorkSpaceSize获取系统workspace的大小。 auto ascendcPlatform platform_ascendc:: PlatformAscendC(context-GetPlatformInfo()); uint32_t sysWorkspaceSize ascendcPlatform.GetLibApiWorkSpaceSize(); // 设置用户需要使用的workspace和GroupBarrier需要的大小作为usrWorkspace的总大小。 size_t usrSize 256 ascendcPlatform.GetResGroupBarrierWorkSpaceSize(); // 设置用户需要使用的workspace大小。 size_t *currentWorkspace context-GetWorkspaceSizes(1); // 通过框架获取workspace的指针GetWorkspaceSizes入参为所需workspace的块数。当前限制使用一块。 currentWorkspace[0] usrSize sysWorkspaceSize; // 设置总的workspace的数值大小总的workspace空间由框架来申请并管理。 ... }【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言原生支持C和C标准规范主要由类库和语言扩展层构成提供多层级API满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考