Only set the hip context if it is valid.

iree-org · Nov 13, 2024 · a48e08f · a48e08f
1 parent f563521
commit a48e08f
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/runtime/src/iree/hal/drivers/hip/context_util.h b/runtime/src/iree/hal/drivers/hip/context_util.h
@@ -13,6 +13,9 @@
 
 static inline iree_status_t iree_hal_hip_set_context(
     const iree_hal_hip_dynamic_symbols_t* syms, hipCtx_t hip_context) {
+  if (!hip_context) {
+    return iree_ok_status();
+  }
   IREE_TRACE({
     hipCtx_t current_context = NULL;
     IREE_HIP_RETURN_IF_ERROR(syms, hipCtxGetCurrent(&current_context),