EngineX-Iluvatar/enginex_bi_series-sherpa-onnx
Archived
15
0
Fork 0
Code Issues Pull Requests Actions Projects Releases Wiki Activity

bi_v100 code change

Browse Source
This commit is contained in:
Lu Xinlong
2025-08-06 10:32:39 +08:00
parent 0d44df9b67
commit f607b98abd
3 changed files with 7 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
cmake/cmake_extension.py
View File

@@ -180,8 +180,8 @@ class BuildExtension(build_ext):
if make_args == "" and system_make_args == "": if make_args == "" and system_make_args == "":
print("for fast compilation, run:") print("for fast compilation, run:")
print('export SHERPA_ONNX_MAKE_ARGS="-j"; python setup.py install') print('export SHERPA_ONNX_MAKE_ARGS="-j"; python setup.py install')
print('Setting make_args to "-j4"') print('Setting make_args to "-j8"')
make_args = "-j4" make_args = "-j8"
if "-G Ninja" in cmake_args: if "-G Ninja" in cmake_args:
build_cmd = f""" build_cmd = f"""

8
cmake/onnxruntime-linux-x86_64-gpu.cmake
View File

@@ -19,9 +19,9 @@ if(NOT SHERPA_ONNX_ENABLE_GPU)
endif() 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_URL "ftp://ftp.4pd.io/pub/iluvatar/bi_v100/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_URL2 "ftp://ftp.4pd.io/pub/iluvatar/bi_v100/onnxruntime-linux-x64-gpu-1.17.1-patched.zip")
set(onnxruntime_HASH "SHA256=1261de176e8d9d4d2019f8fa8c732c6d11494f3c6e73168ab6d2cc0903f22551") set(onnxruntime_HASH "SHA256=3b07e07aa756b625892f6933acd93f9606f32347c0342e55690f859975318014")
# If you don't have access to the Internet, # If you don't have access to the Internet,
# please download onnxruntime to one of the following locations. # please download onnxruntime to one of the following locations.
@@ -70,7 +70,7 @@ add_library(onnxruntime SHARED IMPORTED)
set_target_properties(onnxruntime PROPERTIES set_target_properties(onnxruntime PROPERTIES
IMPORTED_LOCATION ${location_onnxruntime} 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 find_library(location_onnxruntime_cuda_lib onnxruntime_providers_cuda

2
setup.py
View File

@@ -31,7 +31,7 @@ def get_package_version():
cmake_args = os.environ.get("SHERPA_ONNX_CMAKE_ARGS", "") cmake_args = os.environ.get("SHERPA_ONNX_CMAKE_ARGS", "")
extra_version = "" extra_version = ""
if "-DSHERPA_ONNX_ENABLE_GPU=ON" in cmake_args: if "-DSHERPA_ONNX_ENABLE_GPU=ON" in cmake_args:
extra_version = "+cuda" extra_version = "+corex3.2.1"
latest_version += extra_version latest_version += extra_version