GetUserWorkspace
功能说明
获取用户使用的workspace指针。workspace的具体介绍请参考workspace。Kernel直调开发方式下,如果未开启-DHAVE_WORKSPACE编译选项,框架不会自动设置系统workspace。如果使用了Matmul等需要系统workspace的高阶API,kernel侧需要通过SetSysWorkSpace设置系统workspace,此时用户workspace需要通过该接口获取。
函数原型
1
|
__aicore__ inline GM_ADDR GetUserWorkspace(GM_ADDR workspace) |
参数说明
参数名称 |
输入/输出 |
描述 |
---|---|---|
workspace |
输入 |
传入workspace的指针,包括系统workspace和用户使用的workspace。 |
支持的型号
Atlas 训练系列产品
Atlas推理系列产品(Ascend 310P处理器)AI Core
Atlas A2训练系列产品/Atlas 800I A2推理产品
注意事项
无
返回值
用户使用workspace指针。
父主题: workspace