昇腾社区首页
中文
注册

CheckCopyValid

函数功能

检测两块TensorBuffer对象的内存长度是否相同,符合相互拷贝条件。

函数原型

static APP_ERROR CheckCopyValid(const TensorBuffer &buffer1, const TensorBuffer &buffer2);

参数说明

参数名

输入/输出

说明

buffer1

输入

TensorBuffer对象。要求保证实际分配的内存大小与size一致。

buffer2

输入

TensorBuffer对象。要求保证实际分配的内存大小与size一致。

返回参数说明

数据结构

说明

TensorBufferCopyType

两块TensorBuffer内存的数据类型。

  • HOST_AND_HOST:两块都为Host侧内存。
  • HOST_AND_DEVICE:一块为Host侧内存,另一块为Device侧。
  • DEVICE_AND_SAME_DEVICE:两块都为Device侧内存,且deviceId相同。
  • DEVICE_AND_DIFF_DEVICE:两块都为Device侧内存,且deviceId不同。