Support Windows arm64 (#911)
This commit is contained in:
91
cmake/onnxruntime-win-arm64.cmake
Normal file
91
cmake/onnxruntime-win-arm64.cmake
Normal file
@@ -0,0 +1,91 @@
|
||||
# Copyright (c) 2022-2024 Xiaomi Corporation
|
||||
message(STATUS "CMAKE_SYSTEM_NAME: ${CMAKE_SYSTEM_NAME}")
|
||||
message(STATUS "CMAKE_SYSTEM_PROCESSOR: ${CMAKE_SYSTEM_PROCESSOR}")
|
||||
message(STATUS "CMAKE_VS_PLATFORM_NAME: ${CMAKE_VS_PLATFORM_NAME}")
|
||||
|
||||
if(NOT CMAKE_SYSTEM_NAME STREQUAL Windows)
|
||||
message(FATAL_ERROR "This file is for Windows only. Given: ${CMAKE_SYSTEM_NAME}")
|
||||
endif()
|
||||
|
||||
if(NOT (CMAKE_VS_PLATFORM_NAME STREQUAL ARM64 OR CMAKE_VS_PLATFORM_NAME STREQUAL arm64))
|
||||
message(FATAL_ERROR "This file is for Windows arm64 only. Given: ${CMAKE_VS_PLATFORM_NAME}")
|
||||
endif()
|
||||
|
||||
if(NOT BUILD_SHARED_LIBS)
|
||||
message(FATAL_ERROR "This file is for building shared libraries. BUILD_SHARED_LIBS: ${BUILD_SHARED_LIBS}")
|
||||
endif()
|
||||
|
||||
set(onnxruntime_URL "https://github.com/microsoft/onnxruntime/releases/download/v1.17.1/onnxruntime-win-arm64-1.17.1.zip")
|
||||
set(onnxruntime_URL2 "https://hub.nuaa.cf/microsoft/onnxruntime/releases/download/v1.17.1/onnxruntime-win-arm64-1.17.1.zip")
|
||||
set(onnxruntime_HASH "SHA256=47782cebcab0fd7a1f0a3f0676b088c1bc0f4fbf21666f6fe57570dc362fa5a8")
|
||||
|
||||
# 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/onnxruntime-win-arm64-1.17.1.zip
|
||||
${CMAKE_SOURCE_DIR}/onnxruntime-win-arm64-1.17.1.zip
|
||||
${CMAKE_BINARY_DIR}/onnxruntime-win-arm64-1.17.1.zip
|
||||
/tmp/onnxruntime-win-arm64-1.17.1.zip
|
||||
)
|
||||
|
||||
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}/lib"
|
||||
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}/include"
|
||||
)
|
||||
|
||||
set_property(TARGET onnxruntime
|
||||
PROPERTY
|
||||
IMPORTED_IMPLIB "${onnxruntime_SOURCE_DIR}/lib/onnxruntime.lib"
|
||||
)
|
||||
|
||||
file(COPY ${onnxruntime_SOURCE_DIR}/lib/onnxruntime.dll
|
||||
DESTINATION
|
||||
${CMAKE_BINARY_DIR}/bin/${CMAKE_BUILD_TYPE}
|
||||
)
|
||||
|
||||
file(GLOB onnxruntime_lib_files "${onnxruntime_SOURCE_DIR}/lib/*.dll")
|
||||
|
||||
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)
|
||||
@@ -87,9 +87,14 @@ function(download_onnxruntime)
|
||||
if(SHERPA_ONNX_ENABLE_GPU)
|
||||
message(FATAL_ERROR "GPU support for Win32 is not supported!")
|
||||
endif()
|
||||
elseif(CMAKE_VS_PLATFORM_NAME STREQUAL ARM64 OR CMAKE_VS_PLATFORM_NAME STREQUAL arm64)
|
||||
# for 64-bit windows (arm64)
|
||||
if(NOT BUILD_SHARED_LIBS)
|
||||
message(FATAL_ERROR "Please pass -DBUILD_SHARED_LIBS=ON to cmake")
|
||||
endif()
|
||||
include(onnxruntime-win-arm64)
|
||||
else()
|
||||
# for 64-bit windows
|
||||
|
||||
# for 64-bit windows (x64)
|
||||
if(BUILD_SHARED_LIBS)
|
||||
message(STATUS "Use dynamic onnxruntime libraries")
|
||||
if(SHERPA_ONNX_ENABLE_GPU)
|
||||
|
||||
Reference in New Issue
Block a user