Testing dependency download

Author: mc-nv

CMakeLists.txt

@@ -131,14 +131,16 @@ if(NOT CMAKE_BUILD_TYPE)
 endif()
 
 set(TRITON_ONNXRUNTIME_DOCKER_BUILD OFF)
+# Download onnxruntime
+include(cmake/download_onnxruntime.cmake)
 if(TRITON_ONNXRUNTIME_LIB_PATHS STREQUAL "")
   set(TRITON_ONNXRUNTIME_DOCKER_BUILD ON)
 endif()
 
 message(STATUS "Using Onnxruntime docker: ${TRITON_ONNXRUNTIME_DOCKER_BUILD}")
 
 if(NOT TRITON_ONNXRUNTIME_DOCKER_BUILD)
-  find_library(ONNXRUNTIME_LIBRARY NAMES onnxruntime PATHS ${TRITON_ONNXRUNTIME_LIB_PATHS})
+  find_library(ONNXRUNTIME_LIBRARY NAMES onnxruntime PATHS ${TRITON_ONNXRUNTIME_LIB_PATHS} REQUIRED)
   if(${TRITON_ENABLE_ONNXRUNTIME_OPENVINO})
     find_library(OV_LIBRARY
       NAMES openvino

cmake/download_onnxruntime.cmake

@@ -0,0 +1,32 @@
+if(WIN32)
+
+  message(NOTICE "Downloading onnxruntime")
+
+  # Define the download and extraction paths)
+  set(DOWNLOAD_PATH "${CMAKE_BINARY_DIR}/_deps/downloads/onnxruntime.zip")
+  set(EXTRACT_DIR "${CMAKE_BINARY_DIR}/_deps/downloads/onnxruntime")
+
+  # Download the file
+  file(DOWNLOAD $ENV{TRITON_ONNXRUNTIME_PACKAGE_URL} ${DOWNLOAD_PATH} SHOW_PROGRESS STATUS DOWNLOAD_STATUS)
+
+  # Check the download status
+  list(GET DOWNLOAD_STATUS 0 DOWNLOAD_RESULT)
+  if(NOT DOWNLOAD_RESULT EQUAL 0)
+      message(SEND_ERROR "Failed to download ${ONNXRUNTIME_PACKAGE_URL}")
+  else()
+      message(NOTICE "Download successful: ${DOWNLOAD_PATH}" )
+
+      # Extract the downloaded file
+      file(ARCHIVE_EXTRACT INPUT ${DOWNLOAD_PATH} DESTINATION ${EXTRACT_DIR} VERBOSE )
+
+      # Make the extracted directory searchable for CMake
+      list(APPEND CMAKE_PREFIX_PATH ${EXTRACT_DIR})
+
+      # Find a specific library in the extracted folder
+      # find_library(ONNXRUNTIME_LIBRARY onnxruntime NAMES onnxruntime.dll onnxruntime.so PATHS ${EXTRACT_DIR} PATH_SUFFIXES lib include NO_DEFAULT_PATH)
+
+      set(TRITON_ONNXRUNTIME_LIB_PATHS ${EXTRACT_DIR}/lib)
+
+  endif(NOT DOWNLOAD_RESULT EQUAL 0)
+
+endif(WIN32)