昇腾社区首页
中文
注册

aclmdlRICaptureBegin

产品支持情况

产品

是否支持

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

Atlas 200I/500 A2 推理产品

x

Atlas 推理系列产品

x

Atlas 训练系列产品

x

Atlas 200/300/500 推理产品

x

功能说明

开始捕获Stream上下发的任务。

在aclmdlRICaptureBegin和aclmdlRICaptureEnd接口之间,所有在指定Stream上下发的任务不会立即执行,而是被暂存在系统内部模型中,只有在调用aclmdlRIExecuteAsync接口执行模型推理时,这些任务才会被真正执行,以此减少Host侧的任务下发开销。任务执行完毕后,若无需再使用内部模型,可调用aclmdlRIDestroy接口及时销毁该资源。

aclmdlRICaptureBegin和aclmdlRICaptureEnd接口要成对使用,且两个接口中的Stream应相同。在这两个接口之间,可以调用aclmdlRICaptureGetInfo接口获取捕获信息,调用aclmdlRICaptureThreadExchangeMode接口切换当前线程的捕获模式。此外,在调用aclmdlRICaptureEnd接口之后,还可以调用aclmdlRIDebugPrint接口打印模型信息,这在维护和测试场景下有助于问题定位。

函数原型

aclError aclmdlRICaptureBegin(aclrtStream stream, const aclmdlRICaptureMode mode)

参数说明

参数名

输入/输出

说明

stream

输入

指定Stream。

mode

输入

捕获模式,用于限制非安全函数(包括aclrtMemset、aclrtMemcpy、aclrtMemcpy2d)的调用范围。

返回值说明

返回0表示成功,返回其他值表示失败,请参见aclError