From a2fededfda1b800ae5dbba635661e0aa4f5cd617 Mon Sep 17 00:00:00 2001
From: "ysh329@sina.com" <ysh329@sina.com>
Date: Fri, 27 Mar 2020 07:02:52 +0000
Subject: [PATCH] add support CopyFromCpu / CopyToCpu. test=develop

---
 lite/api/paddle_api.cc | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/lite/api/paddle_api.cc b/lite/api/paddle_api.cc
index 2cb2064da5..dd9c43972e 100644
--- a/lite/api/paddle_api.cc
+++ b/lite/api/paddle_api.cc
@@ -126,11 +126,13 @@ template void Tensor::CopyFromCpu<int, TargetType::kHost>(const int *);
 template void Tensor::CopyFromCpu<float, TargetType::kHost>(const float *);
 template void Tensor::CopyFromCpu<int8_t, TargetType::kHost>(const int8_t *);
 template void Tensor::CopyFromCpu<uint8_t, TargetType::kHost>(const uint8_t *);
+template void Tensor::CopyFromCpu<int64_t, TargetType::kHost>(const int64_t *);
 
 template void Tensor::CopyFromCpu<int, TargetType::kARM>(const int *);
 template void Tensor::CopyFromCpu<float, TargetType::kARM>(const float *);
 template void Tensor::CopyFromCpu<int8_t, TargetType::kARM>(const int8_t *);
 template void Tensor::CopyFromCpu<uint8_t, TargetType::kARM>(const uint8_t *);
+template void Tensor::CopyFromCpu<int64_t, TargetType::kARM>(const int64_t *);
 
 template void Tensor::CopyFromCpu<int, TargetType::kCUDA>(const int *);
 template void Tensor::CopyFromCpu<int64_t, TargetType::kCUDA>(const int64_t *);
@@ -141,6 +143,7 @@ template void Tensor::CopyToCpu(float *) const;
 template void Tensor::CopyToCpu(int *) const;
 template void Tensor::CopyToCpu(int8_t *) const;
 template void Tensor::CopyToCpu(uint8_t *) const;
+template void Tensor::CopyToCpu(int64_t *) const;
 
 shape_t Tensor::shape() const {
   return ctensor(raw_tensor_)->dims().Vectorize();
-- 
GitLab