update

PaddlePaddle · Jul 23, 2022 · c24b13e · c24b13e
1 parent cea6739
commit c24b13e
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 5 deletions.
diff --git a/paddle/fluid/inference/api/resource_manager.cc b/paddle/fluid/inference/api/resource_manager.cc
@@ -434,10 +434,12 @@ void GPUContextResource::ReBindSparseHandle(gpuStream_t stream) const {
 
 void GPUContextResource::ReBindEigenDevice(gpuStream_t stream,
                                            GPUPlace place) const {
-  auto* allocator = paddle::memory::allocation::AllocatorFacade::Instance()
-                        .GetAllocator(place_)
-                        .get();
-  eigen_stream_->Reinitialize(stream, allocator, place);
+  if (eigen_stream_) {
+    auto* allocator = paddle::memory::allocation::AllocatorFacade::Instance()
+                          .GetAllocator(place_)
+                          .get();
+    eigen_stream_->Reinitialize(stream, allocator, place);
+  }
 }
 
 #endif

diff --git a/paddle/phi/backends/gpu/gpu_context.cc b/paddle/phi/backends/gpu/gpu_context.cc
@@ -319,7 +319,7 @@ struct GPUContext::Impl {
     eigen_device_creator_ = std::move(creator);
   }
 
-  Eigen::GpuDevice* eigen_device() const {
+  Eigen::GpuDevice* eigen_device() {
     std::call_once(flag_eigen_device_, [&]() {
       if (!eigen_device_) {
         if (!eigen_device_creator_)