From 4a53451c673dd8db790df24dcdc5b7a23a201665 Mon Sep 17 00:00:00 2001 From: Vladislav Perevezentsev Date: Fri, 17 May 2024 15:11:09 +0200 Subject: [PATCH] Wait for each host_task before calling _heevd --- dpnp/linalg/dpnp_utils_linalg.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/dpnp/linalg/dpnp_utils_linalg.py b/dpnp/linalg/dpnp_utils_linalg.py index 80c643731f23..22aa396c7fed 100644 --- a/dpnp/linalg/dpnp_utils_linalg.py +++ b/dpnp/linalg/dpnp_utils_linalg.py @@ -147,12 +147,12 @@ def _batched_eigh(a, UPLO, eigen_mode, w_type, v_type): ) # TODO: Remove this w/a when MKLD-17201 is solved. - # Waiting for a host task executing an OneMKL LAPACK syevd call + # Waiting for a host task executing an OneMKL LAPACK syevd/heevd call # on CPU causes deadlock due to serialization of all host tasks # in the queue. - # We need to wait for each host tasks before calling _seyvd + # We need to wait for each host tasks before calling _seyvd and _heevd # to avoid deadlock. - if lapack_func == "_syevd" and is_cpu_device: + if is_cpu_device: ht_list_ev[2 * i].wait() # call LAPACK extension function to get eigenvalues and