hide SimpleExecutor

Change-Id: I08245abbb5c3fdba91ef1bb0a24871d41594c1ce

paddle/fluid/train/custom_trainer/feed/accessor/epoch_accessor.cc

-#pragma once
 #include "paddle/fluid/train/custom_trainer/feed/accessor/epoch_accessor.h"
 
 namespace paddle {

paddle/fluid/train/custom_trainer/feed/common/registerer.h

@@ -106,7 +106,7 @@ BaseClassMap& global_factory_map_cpp();
     void register_factory_##name() __attribute__((constructor)); 
 
 #define CREATE_CLASS(base_class, name) \
-    base_class##Registerer::CreateInstanceByName(name);
+    base_class##Registerer::CreateInstanceByName(name)
     
 }//namespace feed
 }//namespace custom_trainer

paddle/fluid/train/custom_trainer/feed/executor/executor.cc

@@ -43,26 +43,12 @@ std::unique_ptr<paddle::framework::ProgramDesc> Load(
 
 }
 
-struct SimpleExecutor::Context {
-    Context(const ::paddle::platform::Place& place) : place(place), executor(place) {
-    }
-    const ::paddle::platform::Place& place;
-    ::paddle::framework::Executor executor;
-    ::std::unique_ptr<::paddle::framework::ProgramDesc> main_program;
-    ::std::unique_ptr<framework::ExecutorPrepareContext> prepare_context;
-    details::TensorArrayBatchCleaner tensor_array_batch_cleaner;
-};
-
-
-SimpleExecutor::SimpleExecutor() {
-
-}
-
-SimpleExecutor::~SimpleExecutor() {
-
-}
 
-int SimpleExecutor::initialize(YAML::Node exe_config,
+class SimpleExecutor : public Executor {
+public:
+    SimpleExecutor() {};
+    virtual ~SimpleExecutor() {};
+    virtual int initialize(YAML::Node exe_config,
         std::shared_ptr<TrainerContext> context_ptr) {
         
         paddle::framework::InitDevices(false);
@@ -102,9 +88,8 @@ int SimpleExecutor::initialize(YAML::Node exe_config,
         }
 
         return 0;
-}
-
-int SimpleExecutor::run() {
+    }
+    virtual int run() {
         if (_context == nullptr) {
             VLOG(2) << "need initialize before run";
             return -1;
@@ -122,7 +107,20 @@ int SimpleExecutor::run() {
             return -1;
         }
         return 0;
-}
+    }
+protected:
+    struct Context {
+        Context(const ::paddle::platform::Place& place) : place(place), executor(place) {
+        }
+        const ::paddle::platform::Place& place;
+        ::paddle::framework::Executor executor;
+        ::std::unique_ptr<::paddle::framework::ProgramDesc> main_program;
+        ::std::unique_ptr<framework::ExecutorPrepareContext> prepare_context;
+        details::TensorArrayBatchCleaner tensor_array_batch_cleaner;
+    };
+    std::unique_ptr<Context> _context;
+};
+
 REGISTER_CLASS(Executor, SimpleExecutor);
     
 }  // namespace feed

paddle/fluid/train/custom_trainer/feed/executor/executor.h

@@ -42,18 +42,6 @@ protected:
 };
 REGISTER_REGISTERER(Executor);
 
-class SimpleExecutor : public Executor {
-public:
-    SimpleExecutor();
-    virtual ~SimpleExecutor();
-    virtual int initialize(YAML::Node exe_config,
-        std::shared_ptr<TrainerContext> context_ptr);
-    virtual int run();
-protected:
-    struct Context;
-    std::unique_ptr<Context> _context;
-};
-
 }  // namespace feed
 }  // namespace custom_trainer
 }  // namespace paddle

paddle/fluid/train/custom_trainer/feed/unit_test/test_executor.cc

@@ -81,22 +81,23 @@ public:
 };
 
 TEST_F(SimpleExecutorTest, initialize) {
-    SimpleExecutor executor;
+    std::unique_ptr<Executor> executor(CREATE_CLASS(Executor, "SimpleExecutor"));
     YAML::Node config = YAML::Load("[1, 2, 3]");
-    ASSERT_NE(0, executor.initialize(config, context_ptr));
+    ASSERT_NE(0, executor->initialize(config, context_ptr));
     config = YAML::Load(std::string() + "{startup_program: " + startup_program_path + ", main_program: " + main_program_path + "}");
-    ASSERT_EQ(0, executor.initialize(config, context_ptr));
+    ASSERT_EQ(0, executor->initialize(config, context_ptr));
     config = YAML::Load(std::string() + "{thread_num: 2, startup_program: " + startup_program_path + ", main_program: " + main_program_path + "}");
-    ASSERT_EQ(0, executor.initialize(config, context_ptr));
+    ASSERT_EQ(0, executor->initialize(config, context_ptr));
 }
 
 TEST_F(SimpleExecutorTest, run) {
-    SimpleExecutor executor;
+    std::unique_ptr<Executor> executor(CREATE_CLASS(Executor, "SimpleExecutor"));
+
     auto config = YAML::Load(std::string() + "{thread_num: 2, startup_program: " + startup_program_path + ", main_program: " + main_program_path + "}");
-    ASSERT_EQ(0, executor.initialize(config, context_ptr));
+    ASSERT_EQ(0, executor->initialize(config, context_ptr));
     
-	auto x_var = executor.mutable_var<::paddle::framework::LoDTensor>("x");
-    executor.mutable_var<::paddle::framework::LoDTensor>("mean");
+	auto x_var = executor->mutable_var<::paddle::framework::LoDTensor>("x");
+    executor->mutable_var<::paddle::framework::LoDTensor>("mean");
     ASSERT_NE(nullptr, x_var);
 
     int x_len = 10;
@@ -109,9 +110,9 @@ TEST_F(SimpleExecutorTest, run) {
     }
     std::cout << std::endl;
 
-    ASSERT_EQ(0, executor.run());
+    ASSERT_EQ(0, executor->run());
 
-	auto mean_var = executor.var<::paddle::framework::LoDTensor>("mean");
+	auto mean_var = executor->var<::paddle::framework::LoDTensor>("mean");
     auto mean = mean_var.data<float>()[0];
     std::cout << "mean: " << mean << std::endl;
     ASSERT_NEAR(4.5, mean, 1e-9);