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 {
99
99
#if defined(PADDLE_WITH_DISTRIBUTE) && defined(PADDLE_WITH_PSCORE) && \
100
100
!defined (PADDLE_WITH_ASCEND_CL)
101
101
if (UNLIKELY (config_.dist_config ().use_dist_model ())) {
102
+ VLOG (3 ) << " Using DistModel to inf." ;
102
103
distributed::DistModelConfig dist_model_config;
103
104
dist_model_config.trainer_endpoints =
104
105
config_.dist_config ().trainer_endpoints ();
@@ -464,8 +465,12 @@ class AnalysisPredictor : public PaddlePredictor {
464
465
std::map<std::string, std::vector<std::vector<int32_t >>> shape_info_;
465
466
int clone_num_{1 };
466
467
468
+ #if defined(PADDLE_WITH_DISTRIBUTE) && defined(PADDLE_WITH_PSCORE) && \
469
+ !defined(PADDLE_WITH_ASCEND_CL)
470
+
467
471
private:
468
472
std::unique_ptr<distributed::DistModel> dist_model_{nullptr };
473
+ #endif
469
474
};
470
475
471
476
} // namespace paddle
You can’t perform that action at this time.
0 commit comments