开发者
下载

aclrtFunctionGetAvailDynUbufPerBlock

产品支持情况

产品

是否支持

Ascend 950PR/Ascend 950DT

Atlas A3 训练系列产品/Atlas A3 推理系列产品

Atlas A2 训练系列产品/Atlas A2 推理系列产品

Atlas 200I/500 A2 推理产品

Atlas 推理系列产品

Atlas 训练系列产品

功能说明

对于SIMT(Single Instruction Multiple Thread)核函数,本接口通过dynamicUbufSize输出Launch Kernel前可设置的UB(Unified Buffer,统一缓冲区)最大动态内存大小。

对于非SIMT核函数,本接口返回成功,且dynamicUbufSize固定为0。

函数原型

1
aclError aclrtFunctionGetAvailDynUbufPerBlock(void *func, uint32_t flags, size_t *dynamicUbufSize)

参数说明

参数名

输入/输出

说明

func

输入

核函数句柄。类型定义请参见aclrtFuncHandle

flags

输入

预留参数,当前固定配置为0。

dynamicUbufSize

输出

每个Block可用的UB(Unified Buffer,统一缓冲区)最大动态内存大小,单位为Byte。

返回值说明

返回0表示成功,返回其他值表示失败,请参见aclError