Skip to content

Commit 42ecc88

Browse files
authored
fix mkl path not set error when compiling cppsdk (#1058)
1 parent 29d31e1 commit 42ecc88

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

setup.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -81,7 +81,7 @@
8181

8282
# build mode
8383
pytorch_install_dir = ''
84-
mkl_install_dir = ''
84+
mkl_install_dir = os.path.abspath(os.path.join(os.path.dirname(sys.executable), ".."))
8585
USE_CXX11_ABI = 0
8686
mode = ''
8787
if len(sys.argv) > 1:
@@ -535,7 +535,6 @@ def run(self):
535535
_install_requirements()
536536

537537
# Find the oneMKL library path
538-
mkl_install_dir = os.path.abspath(os.path.join(os.path.dirname(sys.executable), ".."))
539538
mkl_lib_path = mkl_install_dir + "/lib/"
540539
mkl_include_path = mkl_install_dir + "/include/"
541540

0 commit comments

Comments
 (0)