CANN/asc-devkit获取系统工作空间指针

GetSysWorkSpacePtr【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言原生支持C和C标准规范主要由类库和语言扩展层构成提供多层级API满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit产品支持情况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接口获取。函数原型__aicore__ inline __gm__ uint8_t* __gm__ GetSysWorkSpacePtr()参数说明无约束说明无返回值说明系统workspace指针。调用示例... 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本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言原生支持C和C标准规范主要由类库和语言扩展层构成提供多层级API满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考