From e72f054e2bebe7e3980f672cd7a12806e3073d64 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?S=C3=A9rgio=20Peixoto?= Date: Fri, 20 May 2022 12:12:59 +0100 Subject: [PATCH] Support for finding DLL's Adds support for finding the DLLs published by mkl package --- mkl/__init__.py | 15 +++++++++++++++ 1 file changed, 15 insertions(+) diff --git a/mkl/__init__.py b/mkl/__init__.py index 659b1b9..512ffcc 100644 --- a/mkl/__init__.py +++ b/mkl/__init__.py @@ -24,6 +24,19 @@ # OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. import sys +import os + + +def add_dll_directory(): + """mkl package DLL's need to be available so they can be dynamically linked""" + path = os.path.join(sys.prefix, "Library\\bin") + try: + os.add_dll_directory(path) + except AttributeError: + environ_path = os.environ.get('PATH', '') + if path not in environ_path: + os.environ['PATH'] = os.pathsep.join((path, environ_path)) + class RTLD_for_MKL(): def __init__(self): @@ -45,6 +58,8 @@ def __exit__(self, *args): sys.setdlopenflags(self.saved_rtld) self.saved_rtld = None +add_dll_directory() + with RTLD_for_MKL(): from . import _mklinit