File tree Expand file tree Collapse file tree 2 files changed +11
-3
lines changed Expand file tree Collapse file tree 2 files changed +11
-3
lines changed Original file line number Diff line number Diff line change @@ -1032,9 +1032,9 @@ Stub::~Stub()
10321032
10331033  {
10341034    py::gil_scoped_acquire acquire;
1035-     async_event_loop_ =  py::none  ( );
1036-     background_futures_ =  py::none  ( );
1037-     model_instance_ =  py::none  ( );
1035+     py::object  async_event_loop_local ( std::move (async_event_loop_) );
1036+     py::object  background_futures_local ( std::move (background_futures_) );
1037+     py::object  model_instance_local ( std::move (model_instance_) );
10381038  }
10391039  stub_instance_.reset ();
10401040  stub_message_queue_.reset ();
Original file line number Diff line number Diff line change @@ -503,6 +503,14 @@ PbTensor::~PbTensor() noexcept(false)
503503{
504504  pb_memory_.reset ();
505505  DeleteDLPack ();
506+ 
507+ #ifdef  TRITON_PB_STUB
508+   {
509+     py::gil_scoped_acquire acquire;
510+     py::array numpy_array_local (std::move (numpy_array_));
511+     py::array numpy_array_serialized_local (std::move (numpy_array_serialized_));
512+   }
513+ #endif 
506514}
507515
508516const  std::string&
 
 
   
 
     
   
   
          
    
    
     
    
      
     
     
    You can’t perform that action at this time.
  
 
    
  
    
      
        
     
       
      
     
   
 
    
    
  
 
  
 
     
    
0 commit comments