From e47c27358e0afdaf11a42a841ea5c828614066d0 Mon Sep 17 00:00:00 2001
From: zhaojiaying01 <zhaojiaying01@baidu.com>
Date: Wed, 17 Oct 2018 16:55:06 +0800
Subject: [PATCH] add PADDLE_MOBILE_CL macro in fetch_op.h

---
 src/operators/fetch_op.h | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

diff --git a/src/operators/fetch_op.h b/src/operators/fetch_op.h
index 59b9c46ed3..4e313092cc 100644
--- a/src/operators/fetch_op.h
+++ b/src/operators/fetch_op.h
@@ -24,6 +24,7 @@ namespace operators {
 using std::string;
 
 template <typename DeviceType, typename T>
+#ifdef PADDLE_MOBILE_CL
 class FetchOp
     : public framework::OperatorWithKernel<DeviceType, FetchParam<DeviceType>,
                                            FetchKernel<DeviceType, T>> {
@@ -34,12 +35,26 @@ class FetchOp
       : framework::OperatorWithKernel<DeviceType, FetchParam<DeviceType>,
                                       FetchKernel<DeviceType, T>>(
             type, inputs, outputs, attrs, scope) {}
+#else
+class FetchOp : public framework::OperatorBase<DeviceType> {
+ public:
+  FetchOp(const string &type, const VariableNameMap &inputs,
+          const VariableNameMap &outputs, const framework::AttributeMap attrs,
+          std::shared_ptr<framework::Scope> scope)
+      : framework::OperatorBase<DeviceType>(type, inputs, outputs, attrs,
+                                            scope),
+        param_(inputs, outputs, attrs, *scope) {}
 
+  void Init() {}
+#endif
   void InferShape() const override;
 
   void RunImpl() override;
 
  protected:
+#ifndef PADDLE_MOBILE_CL
+  FetchParam<DeviceType> param_;
+#endif
 };
 
 }  // namespace operators
-- 
GitLab