diff options
-rw-r--r-- | tensorflow/core/common_runtime/gpu/gpu_process_state.cc | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/core/common_runtime/gpu/gpu_process_state.cc b/tensorflow/core/common_runtime/gpu/gpu_process_state.cc index 607cb33b80..b18688174d 100644 --- a/tensorflow/core/common_runtime/gpu/gpu_process_state.cc +++ b/tensorflow/core/common_runtime/gpu/gpu_process_state.cc @@ -68,6 +68,7 @@ GPUProcessState* GPUProcessState::instance_ = nullptr; GPUProcessState::GPUProcessState() : gpu_device_enabled_(false) { CHECK(instance_ == nullptr); + instance_ = this; process_state_ = ProcessState::singleton(); } |