Do not use event manager for w/a

IntelPython · Jan 17, 2025 · 7e988da · 7e988da
1 parent a930274
commit 7e988da
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/dpnp/linalg/dpnp_utils_linalg.py b/dpnp/linalg/dpnp_utils_linalg.py
@@ -397,13 +397,14 @@ def _batched_qr(a, mode="reduced"):
         batch_size,
         depends=[copy_ev],
     )
-    _manager.add_event_pair(ht_ev, geqrf_ev)
 
     # w/a to avoid raice conditional on CUDA during multiple runs
     # TODO: Remove it ones the OneMath issue is resolved
     # https://github.com/uxlfoundation/oneMath/issues/626
     if dpnp.is_cuda_backend(a_sycl_queue):
         ht_ev.wait()
+    else:
+        _manager.add_event_pair(ht_ev, geqrf_ev)
 
     if mode in ["r", "raw"]:
         if mode == "r":
@@ -2474,13 +2475,14 @@ def dpnp_qr(a, mode="reduced"):
     ht_ev, geqrf_ev = li._geqrf(
         a_sycl_queue, a_t.get_array(), tau_h.get_array(), depends=[copy_ev]
     )
-    _manager.add_event_pair(ht_ev, geqrf_ev)
 
     # w/a to avoid raice conditional on CUDA during multiple runs
     # TODO: Remove it ones the OneMath issue is resolved
     # https://github.com/uxlfoundation/oneMath/issues/626
     if dpnp.is_cuda_backend(a_sycl_queue):
         ht_ev.wait()
+    else:
+        _manager.add_event_pair(ht_ev, geqrf_ev)
 
     if mode in ["r", "raw"]:
         if mode == "r":