Skip to content

Commit 31f9405

Browse files
Wait for each host_task before calling _heevd (#1832)
Co-authored-by: Anton <[email protected]>
1 parent f772017 commit 31f9405

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

dpnp/linalg/dpnp_utils_linalg.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -147,12 +147,12 @@ def _batched_eigh(a, UPLO, eigen_mode, w_type, v_type):
147147
)
148148

149149
# TODO: Remove this w/a when MKLD-17201 is solved.
150-
# Waiting for a host task executing an OneMKL LAPACK syevd call
150+
# Waiting for a host task executing an OneMKL LAPACK syevd/heevd call
151151
# on CPU causes deadlock due to serialization of all host tasks
152152
# in the queue.
153-
# We need to wait for each host tasks before calling _seyvd
153+
# We need to wait for each host tasks before calling _seyvd and _heevd
154154
# to avoid deadlock.
155-
if lapack_func == "_syevd" and is_cpu_device:
155+
if is_cpu_device:
156156
ht_list_ev[2 * i].wait()
157157

158158
# call LAPACK extension function to get eigenvalues and

0 commit comments

Comments
 (0)