GetUserWorkspace

产品支持情况

产品

是否支持

Ascend 950PR/Ascend 950DT

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

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

Kirin X90

Kirin 9030

功能说明

获取用户使用的workspace指针。Kernel直调开发方式下,如果未开启HAVE_WORKSPACE编译选项,框架不会自动设置系统workspace。如果使用了Matmul Kernel侧接口等需要系统workspace的高阶API,kernel侧需要通过SetSysWorkSpace设置系统workspace,此时用户workspace需要通过该接口获取。

函数原型

__aicore__ inline GM_ADDR GetUserWorkspace(GM_ADDR workspace)

参数说明

表 1 接口参数说明

参数名称

输入/输出

描述

workspace

输入

传入workspace的指针,包括系统workspace和用户使用的workspace。

约束说明

返回值说明

用户使用workspace指针。