跳到主要内容

SetErrorCode

DefaultStream

函数功能

在当前进程创建一个默认Stream,静态对象,不支持异步调用。使用前需初始化设备并调用SetDevice指定具体设备ID。

函数原型

static AscendStream &DefaultStream();

返回参数说明

数据结构说明
AscendStream返回AscendStream类。

父主题: AscendStream

GetDeviceId

函数功能

获取Stream所在设备ID(deviceId)。

函数原型

int32_t GetDeviceId() const;

返回参数说明

数据结构说明
int32_tStream所在的设备ID。

父主题: AscendStream

SetErrorCode

函数功能

记录异步任务中产生的错误码。

接口将会记录第一次以及最后一次产生的错误码,可通过GetErrorCode接口获取。

函数原型

void SetErrorCode(APP_ERROR errCode);

参数说明

参数名输入/输出说明
errCode输入错误码,记录单次产生的错误码。

父主题: AscendStream

GetErrorCode

函数功能

获取并返回已记录的错误码,以pair(对)形式返回。

SetErrorCode配合使用,需在调用Synchronize之后使用。

函数原型

std::pair<APP_ERROR, APP_ERROR> GetErrorCode();

返回参数说明

数据结构说明
std::pair<APP_ERROR, APP_ERROR>返回已记录的错误码对,其中第一个字段为第一次记录的错误码,第二个字段为最后记录的错误码。

父主题: AscendStream

LaunchCallback

函数功能

提供异步回调功能。

需在调用CreateAscendStream之后使用。

函数原型

APP_ERROR LaunchCallBack(aclrtCallback fn, void* userData)

参数说明

参数名输入/输出说明
fn输入指定要增加的回调函数。回调函数的函数原型为参见如下。
text<br>typedef void (*aclrtCallback)(void *userData)<br>
userData输入回调函数所需的参数。

返回参数说明

数据结构说明
APP_ERROR程序执行返回的错误码,请参考APP_ERROR说明

父主题: AscendStream

AddTensorRefPtr

函数功能

提供异步回调功能。

需在调用CreateAscendStream之后使用。

将输入Tensor挂载到AscendStream对象上,延长输入Tensor的生命周期。当调用Synchronize或者DestroyAscendStream时,AscendStream对象会释放被挂载的Tensor。

函数原型

void AddTensorRefPtr(const Tensor& inputTensor);

参数说明

参数名输入/输出说明
inputTensor输入Tensor类,需要挂载到AscendStream对象上的张量。

父主题: AscendStream

在线提单