diff --git a/cmake/cmake_extension.py b/cmake/cmake_extension.py index 1810ddbd..2c5d95af 100644 --- a/cmake/cmake_extension.py +++ b/cmake/cmake_extension.py @@ -180,8 +180,8 @@ class BuildExtension(build_ext): if make_args == "" and system_make_args == "": print("for fast compilation, run:") print('export SHERPA_ONNX_MAKE_ARGS="-j"; python setup.py install') - print('Setting make_args to "-j4"') - make_args = "-j4" + print('Setting make_args to "-j8"') + make_args = "-j8" if "-G Ninja" in cmake_args: build_cmd = f""" diff --git a/cmake/onnxruntime-linux-x86_64-gpu.cmake b/cmake/onnxruntime-linux-x86_64-gpu.cmake index 7aed2526..34857fe9 100644 --- a/cmake/onnxruntime-linux-x86_64-gpu.cmake +++ b/cmake/onnxruntime-linux-x86_64-gpu.cmake @@ -19,9 +19,9 @@ if(NOT SHERPA_ONNX_ENABLE_GPU) endif() -set(onnxruntime_URL "https://github.com/csukuangfj/onnxruntime-libs/releases/download/v1.17.1/onnxruntime-linux-x64-gpu-1.17.1-patched.zip") -set(onnxruntime_URL2 "https://hf-mirror.com/csukuangfj/onnxruntime-libs/resolve/main/onnxruntime-linux-x64-gpu-1.17.1-patched.zip") -set(onnxruntime_HASH "SHA256=1261de176e8d9d4d2019f8fa8c732c6d11494f3c6e73168ab6d2cc0903f22551") +set(onnxruntime_URL "ftp://ftp.4pd.io/pub/iluvatar/bi_v100/onnxruntime-linux-x64-gpu-1.17.1-patched.zip") +set(onnxruntime_URL2 "ftp://ftp.4pd.io/pub/iluvatar/bi_v100/onnxruntime-linux-x64-gpu-1.17.1-patched.zip") +set(onnxruntime_HASH "SHA256=3b07e07aa756b625892f6933acd93f9606f32347c0342e55690f859975318014") # If you don't have access to the Internet, # please download onnxruntime to one of the following locations. @@ -70,7 +70,7 @@ add_library(onnxruntime SHARED IMPORTED) set_target_properties(onnxruntime PROPERTIES IMPORTED_LOCATION ${location_onnxruntime} - INTERFACE_INCLUDE_DIRECTORIES "${onnxruntime_SOURCE_DIR}/include" + INTERFACE_INCLUDE_DIRECTORIES "${onnxruntime_SOURCE_DIR}/include/onnxruntime" ) find_library(location_onnxruntime_cuda_lib onnxruntime_providers_cuda diff --git a/setup.py b/setup.py index 54a99178..fe6217b8 100644 --- a/setup.py +++ b/setup.py @@ -31,7 +31,7 @@ def get_package_version(): cmake_args = os.environ.get("SHERPA_ONNX_CMAKE_ARGS", "") extra_version = "" if "-DSHERPA_ONNX_ENABLE_GPU=ON" in cmake_args: - extra_version = "+cuda" + extra_version = "+corex3.2.1" latest_version += extra_version