GetUserWorkspace
产品支持情况
- Ascend 950PR/Ascend 950DT:支持
- Atlas A3 训练系列产品/Atlas A3 推理系列产品:支持
- Atlas A2 训练系列产品/Atlas A2 推理系列产品:支持
- Atlas 200I/500 A2 推理产品:不支持
- Atlas 推理系列产品AI Core:支持
- Atlas 推理系列产品Vector Core:不支持
- Atlas 训练系列产品:支持
- Kirin X90:支持
- Kirin 9030:支持
功能说明
获取用户使用的workspace指针。Kernel直调开发方式下,如果未开启HAVE_WORKSPACE编译选项,框架不会自动设置系统workspace。如果使用了Matmul Kernel侧接口等需要系统workspace的高阶API,kernel侧需要通过SetSysWorkSpace设置系统workspace,此时用户workspace需要通过该接口获取。
函数原型
Text
__aicore__ inline GM_ADDR GetUserWorkspace(GM_ADDR workspace)
参数说明
表1 接口参数说明
参数名称 | 输入/输出 | 描述 |
|---|---|---|
workspace | 输入 | 传入workspace的指针,包括系统workspace和用户使用的workspace。 |
约束说明
无
返回值说明
用户使用workspace指针。