diff --git a/paddle/fluid/inference/api/api_anakin_engine.h b/paddle/fluid/inference/api/api_anakin_engine.h index 32f8def63c090be875076a458a291d43142f51ab..0f0d7febe2ed7331830227c95fed31ac496fa129 100644 --- a/paddle/fluid/inference/api/api_anakin_engine.h +++ b/paddle/fluid/inference/api/api_anakin_engine.h @@ -58,18 +58,18 @@ class PaddleInferenceAnakinPredictor : public PaddlePredictor { ~PaddleInferenceAnakinPredictor() override; - static std::mutex mutex_; - AnakinConfig config_; - std::shared_ptr> ctx_p_; - std::shared_ptr> graph_p_; - anakin::Net* executor_p_{nullptr}; - + protected: void InitEnv(); void InitGraph(); virtual void OptimizeGraph(); virtual void InitNet(); virtual void SetContext(); virtual void Predict(); + static std::mutex mutex_; + AnakinConfig config_; + std::shared_ptr> ctx_p_; + std::shared_ptr> graph_p_; + anakin::Net* executor_p_{nullptr}; private: bool RunImpl(const std::vector& inputs,