From 57c62291cdc2e76da7749ba00366ecb939d0a0d1 Mon Sep 17 00:00:00 2001 From: yangfei Date: Mon, 12 Nov 2018 19:14:11 +0800 Subject: [PATCH] add cpu and gpu predict function --- src/io/paddle_test_inference_api.cpp | 2 ++ src/io/paddle_test_inference_api.h | 2 -- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/src/io/paddle_test_inference_api.cpp b/src/io/paddle_test_inference_api.cpp index 1d69856653..c3ce9d6b6a 100644 --- a/src/io/paddle_test_inference_api.cpp +++ b/src/io/paddle_test_inference_api.cpp @@ -13,6 +13,8 @@ See the License for the specific language governing permissions and limitations under the License. */ #include "io/paddle_test_inference_api.h" +#include "common/types.h" +#include "io/paddle_mobile.h" namespace paddle_mobile { template double PaddleTester::CaculatePredictTime(std::string *cl_path) { diff --git a/src/io/paddle_test_inference_api.h b/src/io/paddle_test_inference_api.h index 528647aae0..448f8f78e0 100644 --- a/src/io/paddle_test_inference_api.h +++ b/src/io/paddle_test_inference_api.h @@ -20,8 +20,6 @@ limitations under the License. */ */ #pragma once -#include "common/types.h" -#include "io/paddle_mobile.h" #include "string" namespace paddle_mobile { -- GitLab