GetSysWorkSpacePtr
产 品 支 持 情 况
- 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指 针。部 分 高 阶API如Matmul需 要 使 用 系 统workspace,相 关 接 口 需 要 传 入 系 统workspace指 针,此 时 可 以 通 过 该 接 口 获 取。使 用 系 统workspace时,host侧 开 发 者 需 要 自 行 申 请 系 统workspace的 空 间,其 预 留 空 间 大 小 可 以 通 过GetLibApiWorkSpaceSize接 口 获 取。
函 数 原 型
Text
__aicore__ inline __gm__ uint8_t* __gm__ GetSysWorkSpacePtr()
参 数 说 明
无
约 束 说 明
无
返 回 值 说 明
系 统workspace指 针。
调 用 示 例
Text
...
REGIST_MATMUL_OBJ(&pipe, GetSysWorkSpacePtr(), mm, &tiling); // 初始化,传入系统workspace指针
// CopyIn阶段:完成从GM到LocalMemory的搬运
mm.SetTensorA(gm_a); // 设置左矩阵A
mm.SetTensorB(gm_b); // 设置右矩阵B
mm.SetBias(gm_bias); // 设置Bias
// Compute阶段:完成矩阵乘计算
while (mm.Iterate()) {
// CopyOut阶段:完成从LocalMemory到GM的搬运
mm.GetTensorC(gm_c);
}
// 结束矩阵乘操作
mm.End();