Set mkl path on linux

huggingface · Oct 23, 2023 · 72c6c6b · 72c6c6b
1 parent d4895e0
commit 72c6c6b
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/candle-pyo3/py_src/candle/__init__.py b/candle-pyo3/py_src/candle/__init__.py
@@ -34,8 +34,7 @@ def locate_mkl_dlls():
                     oneapi_root, "compiler", "latest", "windows", "redist", "intel64_win", "compiler"
                 )
             else:
-                # Unsure of this is correct
-                mkl_path = os.path.join(oneapi_root, "mkl", "latest", "lib")
+                mkl_path = os.path.join(oneapi_root, "mkl", "latest", "lib", "intel64")
 
             logging.warning(f"Adding {mkl_path} to DLL search path...")
             os.add_dll_directory(mkl_path)