|
|
|
@ -157,7 +157,10 @@ void TensorCopyAsync(paddle::lite::Tensor* dst, const framework::LoDTensor& src,
|
|
|
|
|
dst->Resize(framework::vectorize(src.dims()));
|
|
|
|
|
const void* src_data = src.data<void>();
|
|
|
|
|
void* dst_data = dst->mutable_data(bytes);
|
|
|
|
|
VLOG(3) << "[CopyAsync fluid -> lite] Bytes = " << bytes << ", src = " << &src
|
|
|
|
|
<< ", dst = " << dst << ", src_type = " << src.type();
|
|
|
|
|
MemoryCopyAsync(dst_place, dst_data, src_place, src_data, bytes, ctx);
|
|
|
|
|
VLOG(3) << "[Lite memory size] Bytes = " << dst->memory_size();
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
template <>
|
|
|
|
@ -172,7 +175,10 @@ void TensorCopyAsync(framework::LoDTensor* dst, const paddle::lite::Tensor& src,
|
|
|
|
|
const void* src_data = src.raw_data();
|
|
|
|
|
// When Lite is ready, the source type needs to be modified here.
|
|
|
|
|
void* dst_data = dst->mutable_data(dst_place, dst->type());
|
|
|
|
|
VLOG(3) << "[CopyAsync lite -> fluid] Bytes = " << bytes << ", src = " << &src
|
|
|
|
|
<< ", dst = " << dst << ", src_type = " << dst->type();
|
|
|
|
|
MemoryCopyAsync(dst_place, dst_data, src_place, src_data, bytes, ctx);
|
|
|
|
|
VLOG(3) << "[Lite memory size] Bytes = " << src.memory_size();
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
} // namespace utils
|
|
|
|
|