文档管理中心

GetSysWorkSpacePtr

功能说明

获取系统workspace指针。部分高阶API如Matmul需要使用系统workspace,相关接口需要传入系统workspace指针,此时可以通过该接口获取。使用系统workspace时,host侧开发者需要自行申请系统workspace的空间,其预留空间大小可以通过GetLibApiWorkSpaceSize接口获取。

函数原型

收起
自动换行
深色代码主题
复制
  1. __aicore__ inline __gm__ uint8_t* __gm__ GetSysWorkSpacePtr()

参数说明

支持的型号

Kirin9020系列处理器

Kirin9030系列处理器

KirinX90系列处理器

注意事项

返回值

系统workspace指针。

调用示例

收起
自动换行
深色代码主题
复制
  1. // ...
  2. REGIST_MATMUL_OBJ(&pipe, GetSysWorkSpacePtr(), mm, &tiling); // 初始化
  3. // CopyIn阶段:完成从GM到LocalMemory的搬运
  4. mm.SetTensorA(gm_a); // 设置左矩阵A
  5. mm.SetTensorB(gm_b); // 设置右矩阵B
  6. mm.SetBias(gm_bias); // 设置Bias
  7. // Compute阶段:完成矩阵乘计算
  8. while (mm.Iterate()) {
  9. // CopyOut阶段:完成从LocalMemory到GM的搬运
  10. mm.GetTensorC(gm_c);
  11. }
  12. // 结束矩阵乘操作
  13. mm.End();
在 开发与测试 开放能力API 中进行搜索
请输入您想要搜索的关键词