fix dml with preinstall ort (#2066)

2025-03-30 12:07:19 +08:00
parent 3420c89883
commit ce196fceae
4 changed files with 95 additions and 72 deletions
--- a/cmake/onnxruntime-win-x64-directml.cmake
+++ b/cmake/onnxruntime-win-x64-directml.cmake
@@ -19,80 +19,86 @@ if(NOT SHERPA_ONNX_ENABLE_DIRECTML)
  message(FATAL_ERROR "This file is for DirectML. Given SHERPA_ONNX_ENABLE_DIRECTML: ${SHERPA_ONNX_ENABLE_DIRECTML}")
 endif()

-set(onnxruntime_URL  "https://globalcdn.nuget.org/packages/microsoft.ml.onnxruntime.directml.1.14.1.nupkg")
-set(onnxruntime_URL2 "https://hf-mirror.com/csukuangfj/sherpa-onnx-cmake-deps/resolve/main/microsoft.ml.onnxruntime.directml.1.14.1.nupkg")
-set(onnxruntime_HASH "SHA256=c8ae7623385b19cd5de968d0df5383e13b97d1b3a6771c9177eac15b56013a5a")
-
-# If you don't have access to the Internet,
-# please download onnxruntime to one of the following locations.
-# You can add more if you want.
-set(possible_file_locations
-    $ENV{HOME}/Downloads/microsoft.ml.onnxruntime.directml.1.14.1.nupkg
-    ${PROJECT_SOURCE_DIR}/microsoft.ml.onnxruntime.directml.1.14.1.nupkg
-    ${PROJECT_BINARY_DIR}/microsoft.ml.onnxruntime.directml.1.14.1.nupkg
-    /tmp/microsoft.ml.onnxruntime.directml.1.14.1.nupkg
-)
-
-foreach(f IN LISTS possible_file_locations)
-  if(EXISTS ${f})
-    set(onnxruntime_URL  "${f}")
-    file(TO_CMAKE_PATH "${onnxruntime_URL}" onnxruntime_URL)
-    message(STATUS "Found local downloaded onnxruntime: ${onnxruntime_URL}")
-    set(onnxruntime_URL2)
-    break()
-  endif()
-endforeach()
-
-FetchContent_Declare(onnxruntime
-  URL
-    ${onnxruntime_URL}
-    ${onnxruntime_URL2}
-  URL_HASH          ${onnxruntime_HASH}
-)
-
-FetchContent_GetProperties(onnxruntime)
-if(NOT onnxruntime_POPULATED)
-  message(STATUS "Downloading onnxruntime from ${onnxruntime_URL}")
-  FetchContent_Populate(onnxruntime)
-endif()
-message(STATUS "onnxruntime is downloaded to ${onnxruntime_SOURCE_DIR}")
-
-find_library(location_onnxruntime onnxruntime
-  PATHS
-  "${onnxruntime_SOURCE_DIR}/runtimes/win-x64/native"
-  NO_CMAKE_SYSTEM_PATH
-)
-
-message(STATUS "location_onnxruntime: ${location_onnxruntime}")
-
-add_library(onnxruntime SHARED IMPORTED)
-
-set_target_properties(onnxruntime PROPERTIES
-  IMPORTED_LOCATION ${location_onnxruntime}
-  INTERFACE_INCLUDE_DIRECTORIES "${onnxruntime_SOURCE_DIR}/build/native/include"
-)
-
-set_property(TARGET onnxruntime
-  PROPERTY
-    IMPORTED_IMPLIB "${onnxruntime_SOURCE_DIR}/runtimes/win-x64/native/onnxruntime.lib"
-)
-
-file(COPY ${onnxruntime_SOURCE_DIR}/runtimes/win-x64/native/onnxruntime.dll
-  DESTINATION
-    ${CMAKE_BINARY_DIR}/bin/${CMAKE_BUILD_TYPE}
-)
-
-file(GLOB onnxruntime_lib_files "${onnxruntime_SOURCE_DIR}/runtimes/win-x64/native/onnxruntime.*")
-
-message(STATUS "onnxruntime lib files: ${onnxruntime_lib_files}")
-
-if(SHERPA_ONNX_ENABLE_PYTHON)
-  install(FILES ${onnxruntime_lib_files} DESTINATION ..)
+if(location_onnxruntime_header_dir AND location_onnxruntime_lib)
+    message("Use preinstall onnxruntime with directml: ${location_onnxruntime_lib}")
 else()
-  install(FILES ${onnxruntime_lib_files} DESTINATION lib)
-endif()

-install(FILES ${onnxruntime_lib_files} DESTINATION bin)
+    set(onnxruntime_URL  "https://globalcdn.nuget.org/packages/microsoft.ml.onnxruntime.directml.1.14.1.nupkg")
+    set(onnxruntime_URL2 "https://hf-mirror.com/csukuangfj/sherpa-onnx-cmake-deps/resolve/main/microsoft.ml.onnxruntime.directml.1.14.1.nupkg")
+    set(onnxruntime_HASH "SHA256=c8ae7623385b19cd5de968d0df5383e13b97d1b3a6771c9177eac15b56013a5a")
+
+    # If you don't have access to the Internet,
+    # please download onnxruntime to one of the following locations.
+    # You can add more if you want.
+    set(possible_file_locations
+        $ENV{HOME}/Downloads/microsoft.ml.onnxruntime.directml.1.14.1.nupkg
+        ${PROJECT_SOURCE_DIR}/microsoft.ml.onnxruntime.directml.1.14.1.nupkg
+        ${PROJECT_BINARY_DIR}/microsoft.ml.onnxruntime.directml.1.14.1.nupkg
+        /tmp/microsoft.ml.onnxruntime.directml.1.14.1.nupkg
+    )
+
+    foreach(f IN LISTS possible_file_locations)
+      if(EXISTS ${f})
+        set(onnxruntime_URL  "${f}")
+        file(TO_CMAKE_PATH "${onnxruntime_URL}" onnxruntime_URL)
+        message(STATUS "Found local downloaded onnxruntime: ${onnxruntime_URL}")
+        set(onnxruntime_URL2)
+        break()
+      endif()
+    endforeach()
+
+    FetchContent_Declare(onnxruntime
+      URL
+        ${onnxruntime_URL}
+        ${onnxruntime_URL2}
+      URL_HASH          ${onnxruntime_HASH}
+    )
+
+    FetchContent_GetProperties(onnxruntime)
+    if(NOT onnxruntime_POPULATED)
+      message(STATUS "Downloading onnxruntime from ${onnxruntime_URL}")
+      FetchContent_Populate(onnxruntime)
+    endif()
+    message(STATUS "onnxruntime is downloaded to ${onnxruntime_SOURCE_DIR}")
+
+    find_library(location_onnxruntime onnxruntime
+      PATHS
+      "${onnxruntime_SOURCE_DIR}/runtimes/win-x64/native"
+      NO_CMAKE_SYSTEM_PATH
+    )
+
+    message(STATUS "location_onnxruntime: ${location_onnxruntime}")
+
+    add_library(onnxruntime SHARED IMPORTED)
+
+    set_target_properties(onnxruntime PROPERTIES
+      IMPORTED_LOCATION ${location_onnxruntime}
+      INTERFACE_INCLUDE_DIRECTORIES "${onnxruntime_SOURCE_DIR}/build/native/include"
+    )
+
+    set_property(TARGET onnxruntime
+      PROPERTY
+        IMPORTED_IMPLIB "${onnxruntime_SOURCE_DIR}/runtimes/win-x64/native/onnxruntime.lib"
+    )
+
+    file(COPY ${onnxruntime_SOURCE_DIR}/runtimes/win-x64/native/onnxruntime.dll
+      DESTINATION
+        ${CMAKE_BINARY_DIR}/bin/${CMAKE_BUILD_TYPE}
+    )
+
+    file(GLOB onnxruntime_lib_files "${onnxruntime_SOURCE_DIR}/runtimes/win-x64/native/onnxruntime.*")
+
+    message(STATUS "onnxruntime lib files: ${onnxruntime_lib_files}")
+
+    if(SHERPA_ONNX_ENABLE_PYTHON)
+      install(FILES ${onnxruntime_lib_files} DESTINATION ..)
+    else()
+      install(FILES ${onnxruntime_lib_files} DESTINATION lib)
+    endif()
+
+    install(FILES ${onnxruntime_lib_files} DESTINATION bin)
+
+endif()

 # Setup DirectML

--- a/cmake/onnxruntime.cmake
+++ b/cmake/onnxruntime.cmake
@@ -154,6 +154,9 @@ if(SHERPA_ONNX_USE_PRE_INSTALLED_ONNXRUNTIME_IF_AVAILABLE)
      set(location_onnxruntime_lib $ENV{SHERPA_ONNXRUNTIME_LIB_DIR}/libonnxruntime.dylib)
    elseif(WIN32)
      set(location_onnxruntime_lib $ENV{SHERPA_ONNXRUNTIME_LIB_DIR}/onnxruntime.lib)
+      if(SHERPA_ONNX_ENABLE_DIRECTML)
+        include(onnxruntime-win-x64-directml)
+      endif()
    else()
      set(location_onnxruntime_lib $ENV{SHERPA_ONNXRUNTIME_LIB_DIR}/libonnxruntime.so)
    endif()