File tree Expand file tree Collapse file tree 1 file changed +5
-0
lines changed
paddle/fluid/inference/api Expand file tree Collapse file tree 1 file changed +5
-0
lines changed Original file line number Diff line number Diff line change @@ -99,6 +99,7 @@ class AnalysisPredictor : public PaddlePredictor {
9999#if defined(PADDLE_WITH_DISTRIBUTE) && defined(PADDLE_WITH_PSCORE) && \
100100 !defined (PADDLE_WITH_ASCEND_CL)
101101 if (UNLIKELY (config_.dist_config ().use_dist_model ())) {
102+ VLOG (3 ) << " Using DistModel to inf." ;
102103 distributed::DistModelConfig dist_model_config;
103104 dist_model_config.trainer_endpoints =
104105 config_.dist_config ().trainer_endpoints ();
@@ -464,8 +465,12 @@ class AnalysisPredictor : public PaddlePredictor {
464465 std::map<std::string, std::vector<std::vector<int32_t >>> shape_info_;
465466 int clone_num_{1 };
466467
468+ #if defined(PADDLE_WITH_DISTRIBUTE) && defined(PADDLE_WITH_PSCORE) && \
469+ !defined (PADDLE_WITH_ASCEND_CL)
470+
467471 private:
468472 std::unique_ptr<distributed::DistModel> dist_model_{nullptr };
473+ #endif
469474};
470475
471476} // namespace paddle
You can’t perform that action at this time.
0 commit comments