Update setup.py.in

intel · Aug 11, 2022 · d7f440e · d7f440e
1 parent 3cf7faa
commit d7f440e
Showing 1 changed file with 8 additions and 8 deletions.
diff --git a/python/setup.py.in b/python/setup.py.in
@@ -28,7 +28,7 @@ import pyarrow as pa
 # pyhdk.
 extra_library_dirs = []
 if "HDK_ROOT_PATH" in os.environ:
-    extra_library_dirs.append(os.path.join(os.environ["HDK_ROOT_PATH"], "lib"))
+    extra_library_dirs.append(os.path.join(os.environ["HDK_ROOT_PATH"], "lib64"))
 
 include_dirs = [
     np.get_include(),
@@ -46,8 +46,8 @@ common = Extension(
     ["@CMAKE_CURRENT_SOURCE_DIR@/pyhdk/_common.pyx"],
     language="c++",
     include_dirs=include_dirs,
-    library_dirs=pa.get_library_dirs() + ["@CMAKE_CURRENT_BINARY_DIR@", "."],
-    runtime_library_dirs=pa.get_library_dirs(),
+    library_dirs=pa.get_library_dirs() + ["@CMAKE_CURRENT_BINARY_DIR@", "."] + extra_library_dirs,
+    runtime_library_dirs=pa.get_library_dirs() + extra_library_dirs,
     libraries=pa.get_libraries() + ["QueryEngine", "ConfigBuilder", "Logger", "Shared"],
     extra_compile_args=["-std=c++17", "-D__STDC_FORMAT_MACROS"],
 )
@@ -57,8 +57,8 @@ execute = Extension(
     ["@CMAKE_CURRENT_SOURCE_DIR@/pyhdk/_execute.pyx"],
     language="c++",
     include_dirs=include_dirs,
-    library_dirs=pa.get_library_dirs() + ["@CMAKE_CURRENT_BINARY_DIR@", "."],
-    runtime_library_dirs=pa.get_library_dirs(),
+    library_dirs=pa.get_library_dirs() + ["@CMAKE_CURRENT_BINARY_DIR@", "."] + extra_library_dirs,
+    runtime_library_dirs=pa.get_library_dirs() + extra_library_dirs,
     libraries=pa.get_libraries() + ["QueryEngine"],
     extra_compile_args=["-std=c++17", "-D__STDC_FORMAT_MACROS"],
 )
@@ -68,7 +68,7 @@ sql = Extension(
     ["@CMAKE_CURRENT_SOURCE_DIR@/pyhdk/_sql.pyx"],
     language="c++",
     include_dirs=include_dirs,
-    library_dirs=["@CMAKE_CURRENT_BINARY_DIR@", "."],
+    library_dirs=["@CMAKE_CURRENT_BINARY_DIR@", "."] + extra_library_dirs,
     libraries=["Calcite", "QueryEngine"],
     extra_compile_args=["-std=c++17", "-D__STDC_FORMAT_MACROS"],
 )
@@ -78,8 +78,8 @@ storage = Extension(
     ["@CMAKE_CURRENT_SOURCE_DIR@/pyhdk/_storage.pyx"],
     language="c++",
     include_dirs=include_dirs,
-    library_dirs=pa.get_library_dirs() + ["@CMAKE_CURRENT_BINARY_DIR@", "."],
-    runtime_library_dirs=pa.get_library_dirs(),
+    library_dirs=pa.get_library_dirs() + ["@CMAKE_CURRENT_BINARY_DIR@", "."] + extra_library_dirs,
+    runtime_library_dirs=pa.get_library_dirs() + extra_library_dirs,
     libraries=pa.get_libraries() + ["ArrowStorage", "SchemaMgr", "DataMgr"],
     extra_compile_args=["-std=c++17", "-D__STDC_FORMAT_MACROS"],
 )