Skip to content
版 本

GetSysWorkSpacePtr

产 品 支 持 情 况

产 品

是 否 支 持

Ascend 950PR/Ascend 950DT

Atlas A3 训 练 系 列 产 品/Atlas A3 推 理 系 列 产 品

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

Atlas 200I/500 A2 推 理 产 品

x

Atlas 推 理 系 列 产 品AI Core

Atlas 推 理 系 列 产 品Vector Core

x

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();

免 责 声 明:本 站 内 容 由 asc-devkit 仓 master 分 支 自 动 编 译 生 成,属 于 持 续 开 发 版 本,可 能 存 在 缺 陷,仅 供 预 览 与 参 考。如 需 稳 定 及 商 用 资 料,请 查 阅 官 方 昇 腾 社 区