PR #15340: Fix an escaped error return in GpuCommandBuffer::Finalize()

Imported from GitHub PR #15340 Copybara import of the project: -- 5f0be15 by Shawn Wang <[email protected]>: Fix an escaped error return in GpuCommandBuffer::Finalize() Merging this change closes #15340 COPYBARA_INTEGRATE_REVIEW=#15340 from shawnwang18:shawnw/fix_2node_error 5f0be15 PiperOrigin-RevId: 659491888
openxla · Aug 5, 2024 · 8c99802 · 8c99802
1 parent f4cb89e
commit 8c99802
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/xla/stream_executor/gpu/gpu_command_buffer.cc b/xla/stream_executor/gpu/gpu_command_buffer.cc
@@ -1006,6 +1006,8 @@ absl::Status GpuCommandBuffer::Finalize() {
       } else {
         TF_RETURN_IF_ERROR(retry);
       }
+    } else {
+      TF_RETURN_IF_ERROR(instantiated);
     }
 
     uint64_t end_nanos = tsl::Env::Default()->NowNanos();