Skip to content

Commit 7e988da

Browse files
Do not use event manager for w/a
1 parent a930274 commit 7e988da

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

dpnp/linalg/dpnp_utils_linalg.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -397,13 +397,14 @@ def _batched_qr(a, mode="reduced"):
397397
batch_size,
398398
depends=[copy_ev],
399399
)
400-
_manager.add_event_pair(ht_ev, geqrf_ev)
401400

402401
# w/a to avoid raice conditional on CUDA during multiple runs
403402
# TODO: Remove it ones the OneMath issue is resolved
404403
# https://github.com/uxlfoundation/oneMath/issues/626
405404
if dpnp.is_cuda_backend(a_sycl_queue):
406405
ht_ev.wait()
406+
else:
407+
_manager.add_event_pair(ht_ev, geqrf_ev)
407408

408409
if mode in ["r", "raw"]:
409410
if mode == "r":
@@ -2474,13 +2475,14 @@ def dpnp_qr(a, mode="reduced"):
24742475
ht_ev, geqrf_ev = li._geqrf(
24752476
a_sycl_queue, a_t.get_array(), tau_h.get_array(), depends=[copy_ev]
24762477
)
2477-
_manager.add_event_pair(ht_ev, geqrf_ev)
24782478

24792479
# w/a to avoid raice conditional on CUDA during multiple runs
24802480
# TODO: Remove it ones the OneMath issue is resolved
24812481
# https://github.com/uxlfoundation/oneMath/issues/626
24822482
if dpnp.is_cuda_backend(a_sycl_queue):
24832483
ht_ev.wait()
2484+
else:
2485+
_manager.add_event_pair(ht_ev, geqrf_ev)
24842486

24852487
if mode in ["r", "raw"]:
24862488
if mode == "r":

0 commit comments

Comments
 (0)