From 2041008a342ff4beae62d689c4cfaf28d722e644 Mon Sep 17 00:00:00 2001 From: Anton Volkov Date: Mon, 12 Jun 2023 11:07:19 -0500 Subject: [PATCH] w/a for MKL calls --- dpnp/__init__.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/dpnp/__init__.py b/dpnp/__init__.py index 9478c3aef6c8..fc3fc5ff33fc 100644 --- a/dpnp/__init__.py +++ b/dpnp/__init__.py @@ -41,6 +41,8 @@ os.add_dll_directory(dpctlpath) os.environ["PATH"] = os.pathsep.join([os.getenv("PATH", ""), mypath, dpctlpath]) +# workaround against hanging in OneMKL calls +os.environ.setdefault('SYCL_QUEUE_THREAD_POOL_SIZE', '6') from dpnp.dpnp_array import dpnp_array as ndarray from dpnp.dpnp_flatiter import flatiter as flatiter