diff --git a/candle-pyo3/py_src/candle/__init__.py b/candle-pyo3/py_src/candle/__init__.py index fe92650d11..38718a46cc 100644 --- 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)