-
Notifications
You must be signed in to change notification settings - Fork 414
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Support static linking onnxruntime lib for 32-bit arm (#401)
- Loading branch information
1 parent
1544a57
commit 2f2d3bb
Showing
5 changed files
with
118 additions
and
7 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
|
@@ -32,16 +32,23 @@ concurrency: | |
jobs: | ||
arm_linux_gnueabihf: | ||
runs-on: ${{ matrix.os }} | ||
name: ${{ matrix.os }} ${{ matrix.lib_type }} | ||
strategy: | ||
fail-fast: false | ||
matrix: | ||
os: [ubuntu-latest] | ||
lib_type: [static, shared] | ||
|
||
steps: | ||
- uses: actions/checkout@v4 | ||
with: | ||
fetch-depth: 0 | ||
|
||
- name: ccache | ||
uses: hendrikmuhs/[email protected] | ||
with: | ||
key: ${{ matrix.os }}-arm-${{ matrix.lib_type }} | ||
|
||
- name: cache-qemu | ||
id: cache-qemu | ||
uses: actions/cache@v3 | ||
|
@@ -107,6 +114,19 @@ jobs: | |
run: | | ||
export PATH=$GITHUB_WORKSPACE/toolchain/bin:$PATH | ||
export CMAKE_CXX_COMPILER_LAUNCHER=ccache | ||
export PATH="/usr/lib/ccache:/usr/local/opt/ccache/libexec:$PATH" | ||
cmake --version | ||
lib_type=${{ matrix.lib_type }} | ||
if [[ $lib_type == "shared" ]]; then | ||
export BUILD_SHARED_LIBS=ON | ||
else | ||
export BUILD_SHARED_LIBS=OFF | ||
fi | ||
./build-arm-linux-gnueabihf.sh | ||
ls -lh build-arm-linux-gnueabihf/bin | ||
|
@@ -125,12 +145,14 @@ jobs: | |
qemu-arm ./build-arm-linux-gnueabihf/bin/sherpa-onnx --help | ||
readelf -d ./build-arm-linux-gnueabihf/bin/sherpa-onnx | ||
- name: Copy files | ||
shell: bash | ||
run: | | ||
SHERPA_ONNX_VERSION=v$(grep "SHERPA_ONNX_VERSION" ./CMakeLists.txt | cut -d " " -f 2 | cut -d '"' -f 2) | ||
dst=sherpa-onnx-${SHERPA_ONNX_VERSION}-linux-arm-gnueabihf | ||
dst=sherpa-onnx-${SHERPA_ONNX_VERSION}-linux-arm-gnueabihf-${{ matrix.lib_type }} | ||
mkdir $dst | ||
cp -a build-arm-linux-gnueabihf/install/bin $dst/ | ||
|
@@ -141,13 +163,21 @@ jobs: | |
tar cjvf ${dst}.tar.bz2 $dst | ||
- uses: actions/upload-artifact@v3 | ||
if: matrix.lib_type == 'shared' | ||
with: | ||
name: sherpa-onnx-linux-arm-gnueabihf-shared | ||
path: sherpa-onnx-*linux-arm-gnueabihf-shared.tar.bz2 | ||
|
||
- uses: actions/upload-artifact@v3 | ||
if: matrix.lib_type == 'static' | ||
with: | ||
path: sherpa-onnx-*linux-arm-gnueabihf.tar.bz2 | ||
name: sherpa-onnx-linux-arm-gnueabihf-static | ||
path: sherpa-onnx-*linux-arm-gnueabihf-static.tar.bz2 | ||
|
||
- name: Release pre-compiled binaries and libs for arm linux gnueabihf | ||
- name: Release pre-compiled binaries and libs for arm linux gnueabihf ${{ matrix.lib_type }} | ||
if: github.repository_owner == 'csukuangfj' || github.repository_owner == 'k2-fsa' && github.event_name == 'push' && contains(github.ref, 'refs/tags/') | ||
uses: svenstaro/upload-release-action@v2 | ||
with: | ||
file_glob: true | ||
overwrite: true | ||
file: sherpa-onnx-*linux-arm-gnueabihf.tar.bz2 | ||
file: sherpa-onnx-*linux-arm-gnueabihf*.tar.bz2 |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,72 @@ | ||
# Copyright (c) 2022-2023 Xiaomi Corporation | ||
message(STATUS "CMAKE_SYSTEM_NAME: ${CMAKE_SYSTEM_NAME}") | ||
message(STATUS "CMAKE_SYSTEM_PROCESSOR: ${CMAKE_SYSTEM_PROCESSOR}") | ||
|
||
if(NOT CMAKE_SYSTEM_NAME STREQUAL Linux) | ||
message(FATAL_ERROR "This file is for Linux only. Given: ${CMAKE_SYSTEM_NAME}") | ||
endif() | ||
|
||
if(NOT (CMAKE_SYSTEM_PROCESSOR STREQUAL arm OR CMAKE_SYSTEM_PROCESSOR STREQUAL armv7l)) | ||
message(FATAL_ERROR "This file is for arm only. Given: ${CMAKE_SYSTEM_PROCESSOR}") | ||
endif() | ||
|
||
if(BUILD_SHARED_LIBS) | ||
message(FATAL_ERROR "This file is for building static libraries. BUILD_SHARED_LIBS: ${BUILD_SHARED_LIBS}") | ||
endif() | ||
|
||
set(onnxruntime_URL "https://github.com/csukuangfj/onnxruntime-libs/releases/download/v1.16.0/onnxruntime-linux-arm-static_lib-1.16.0.zip") | ||
set(onnxruntime_URL2 "https://huggingface.co/csukuangfj/onnxruntime-libs/resolve/main/onnxruntime-linux-arm-static_lib-1.16.0.zip") | ||
set(onnxruntime_HASH "SHA256=d2bf3d69a479ac786bf2f019e720218a74634b73cfda758ba50d0b1adef1c76a") | ||
|
||
# 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-linux-arm-static_lib-1.16.0.zip | ||
${PROJECT_SOURCE_DIR}/onnxruntime-linux-arm-static_lib-1.16.0.zip | ||
${PROJECT_BINARY_DIR}/onnxruntime-linux-arm-static_lib-1.16.0.zip | ||
/tmp/onnxruntime-linux-arm-static_lib-1.16.0.zip | ||
/star-fj/fangjun/download/github/onnxruntime-linux-arm-static_lib-1.16.0.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}") | ||
|
||
# for static libraries, we use onnxruntime_lib_files directly below | ||
include_directories(${onnxruntime_SOURCE_DIR}/include) | ||
|
||
file(GLOB onnxruntime_lib_files "${onnxruntime_SOURCE_DIR}/lib/lib*.a") | ||
|
||
set(onnxruntime_lib_files ${onnxruntime_lib_files} PARENT_SCOPE) | ||
|
||
message(STATUS "onnxruntime lib files: ${onnxruntime_lib_files}") | ||
install(FILES ${onnxruntime_lib_files} DESTINATION lib) |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters