文档
注册
评分
提单
论坛
小AI

GetUserWorkspace

功能说明

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

函数原型

1
__aicore__ inline GM_ADDR GetUserWorkspace(GM_ADDR workspace)

参数说明

表1 接口参数说明

参数名称

输入/输出

描述

workspace

输入

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

支持的型号

Atlas 训练系列产品

Atlas推理系列产品(Ascend 310P处理器)AI Core

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

注意事项

返回值

用户使用workspace指针。

搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词