类说明
CopyBetweenHostDevice
函数功能
将src对象拷贝至dst对象。两块TensorBuffer内存分别处于Host侧和Device侧。
函数原型
static APP_ERROR CopyBetweenHostDevice(TensorBuffer &dst, const TensorBuffer &src);
参数说明
| 参数名 | 输入/输出 | 说明 |
|---|---|---|
| dst | 输入 | 复制后的目标内存。 |
| src | 输入 | 待复制的源内存。 |
返回参数说明
| 数据结构 | 说明 |
|---|---|
| APP_ERROR | 程序执行返回的错误码，请参考APP_ERROR说明。 |
父主题： TensorBuffer
CopyBetweenSameDevice
函数功能
将Device侧的src对象拷贝至Device侧的dst对象。两块TensorBuffer内存的deviceId相同。
函数原型
static APP_ERROR CopyBetweenSameDevice(TensorBuffer &dst, const TensorBuffer &src);
参数说明
| 参数名 | 输入/输出 | 说明 |
|---|---|---|
| dst | 输入 | 复制后的目标内存。 |
| src | 输入 | 待复制的源内存。 |
返回参数说明
| 数据结构 | 说明 |
|---|---|
| APP_ERROR | 程序执行返回的错误码，请参考APP_ERROR说明。 |
父主题： TensorBuffer
CopyBetweenDiffDevice
函数功能
将Device侧的src对象拷贝至Device侧的dst对象。两块TensorBuffer内存的deviceId不同。
函数原型
static APP_ERROR CopyBetweenDiffDevice(TensorBuffer &dst, const TensorBuffer &src);
参数说明
| 参数名 | 输入/输出 | 说明 |
|---|---|---|
| dst | 输入 | 复制后的目标内存。 |
| src | 输入 | 待复制的源内存。 |
返回参数说明
| 数据结构 | 说明 |
|---|---|
| APP_ERROR | 程序执行返回的错误码，请参考APP_ERROR说明。 |
父主题： TensorBuffer
类说明
该类主要有TensorContext数据结构的构造方法和相关的功能接口。
父主题： TensorContext
TensorContext
函数功能
TensorContext构造函数，创建TensorContext()对象并初始化AscendCL。
函数原型
TensorContext();
父主题： TensorContext
~TensorContext
函数功能
TensorContext类的默认析构函数。
函数原型
~TensorContext();
父主题： TensorContext
GetInstance
函数功能
获取TensorContext单例对象。
函数原型
static std::shared_ptr<TensorContext> GetInstance();
返回参数说明
| 数据结构 | 说明 |
|---|---|
| std::shared_ptr | TensorContext单例对象。 |
父主题： TensorContext
SetContext
函数功能
设置线程的Context。
函数原型
APP_ERROR SetContext(const uint32_t &deviceId);
参数说明
| 参数名 | 输入/输出 | 说明 |
|---|---|---|
| deviceId | 输入 | 线程当前的deviceId。 |
返回参数说明
| 数据结构 | 说明 |
|---|---|
| APP_ERROR | 程序执行返回的错误码，请参考APP_ERROR说明。 |
父主题： TensorContext
在线提单