diff --git a/CMakeLists.txt b/CMakeLists.txt index 94d396d..3e450fd 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -418,7 +418,7 @@ install( # that tar file. We copy over the libraries and other requirements # prior to running this build and therefore these set of install # commands are not needed. -if(TRITON_ONNXRUNTIME_DOCKER_BUILD) +if(TRITON_ONNXRUNTIME_DOCKER_BUILD OR DEFINED TRITON_ONNXRUNTIME_PACKAGE_URL) install( DIRECTORY ${CMAKE_CURRENT_BINARY_DIR}/onnxruntime/ diff --git a/cmake/download_onnxruntime.cmake b/cmake/download_onnxruntime.cmake index 0c09025..6401c2b 100644 --- a/cmake/download_onnxruntime.cmake +++ b/cmake/download_onnxruntime.cmake @@ -1,7 +1,7 @@ if(DEFINED TRITON_ONNXRUNTIME_PACKAGE_URL) set(DOWNLOAD_PATH "${CMAKE_BINARY_DIR}/_deps/downloads/onnxruntime.zip") - set(EXTRACT_DIR "${CMAKE_BINARY_DIR}/_deps/downloads/onnxruntime") + set(EXTRACT_DIR "${CMAKE_BINARY_DIR}/onnxruntime") message(NOTICE "Downloading onnxruntime: ${TRITON_ONNXRUNTIME_PACKAGE_URL}")