Fixed a double Notify() in GpuDriver::LoadPtx

PiperOrigin-RevId: 608969995
tensorflow · Feb 21, 2024 · b18244e · b18244e
1 parent 6e99901
commit b18244e
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/third_party/xla/xla/stream_executor/cuda/cuda_driver.cc b/third_party/xla/xla/stream_executor/cuda/cuda_driver.cc
@@ -1396,6 +1396,7 @@ struct BitPatternToValue {
             "Failed to load PTX text as a module: %s", ToString(res)));
       }
       notification.Notify();
+      return;
     }
 
     VLOG(3) << "PTX compilation info log (" << info_log_buffer_bytes