CopyToNpu
函数功能
创建一个host侧到device侧的数据拷贝任务,并放入executor的任务队列中。
函数原型
const aclTensor *CopyToNpu(const aclTensor *src, aclOpExecutor *executor)
参数说明
参数 |
输入/输出 |
说明 |
---|---|---|
src |
输入 |
需要拷贝到device侧的host侧数据。 |
executor |
输入 |
L2接口中一阶段接口声明的算子执行器对象。 |
返回值说明
拷贝到device侧后,指向device侧数据的aclTensor。任务创建失败则返回nullptr。
约束说明
入参指针不能为空。
调用示例
// 初始化一个host侧tensor,并拷贝到dst,dst为一个device侧tensor void Func(aclOpExecutor *executor) { int64_t myArray[10]; auto src = executor->ConvertToTensor(myArray, 10, DT_INT64); auto dst = CopyToNpu(src, executor); }
父主题: framework_op