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