Skip to content

Commit fd6d650

Browse files
committed
fix compile error
1 parent 852c91f commit fd6d650

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

paddle/fluid/inference/api/analysis_predictor.h

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff 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

0 commit comments

Comments
 (0)