Support RISC-V (#609)

2024-02-26 06:57:18 +08:00
parent 67acd34dcd
commit ee37d9bd92
7 changed files with 437 additions and 1 deletions
--- a/cmake/onnxruntime.cmake
+++ b/cmake/onnxruntime.cmake
@@ -6,6 +6,12 @@ function(download_onnxruntime)
  message(STATUS "CMAKE_SYSTEM_PROCESSOR: ${CMAKE_SYSTEM_PROCESSOR}")
  if(SHERPA_ONNX_ENABLE_WASM)
      include(onnxruntime-wasm-simd)
+  elseif(CMAKE_SYSTEM_PROCESSOR STREQUAL riscv64)
+    if(BUILD_SHARED_LIBS)
+      include(onnxruntime-linux-riscv64)
+    else()
+      include(onnxruntime-linux-riscv64-static)
+    endif()
  elseif(CMAKE_SYSTEM_NAME STREQUAL Linux AND CMAKE_SYSTEM_PROCESSOR STREQUAL aarch64)
    if(BUILD_SHARED_LIBS)
      include(onnxruntime-linux-aarch64)