Просмотр исходного кода

wasi-nn: fix shared library filenames for macOS (#4306)

tested with openvino
YAMAMOTO Takashi 7 месяцев назад
Родитель
Сommit
61cb97221e
1 измененных файлов с 8 добавлено и 3 удалено
  1. 8 3
      core/iwasm/libraries/wasi-nn/src/wasi_nn.c

+ 8 - 3
core/iwasm/libraries/wasi-nn/src/wasi_nn.c

@@ -21,9 +21,14 @@
 #include "wasm_export.h"
 
 #define HASHMAP_INITIAL_SIZE 20
-#define TFLITE_BACKEND_LIB "libwasi_nn_tflite.so"
-#define OPENVINO_BACKEND_LIB "libwasi_nn_openvino.so"
-#define LLAMACPP_BACKEND_LIB "libwasi_nn_llamacpp.so"
+#if defined(__APPLE__)
+#define LIB_EXTENTION ".dylib"
+#else
+#define LIB_EXTENTION ".so"
+#endif
+#define TFLITE_BACKEND_LIB "libwasi_nn_tflite" LIB_EXTENTION
+#define OPENVINO_BACKEND_LIB "libwasi_nn_openvino" LIB_EXTENTION
+#define LLAMACPP_BACKEND_LIB "libwasi_nn_llamacpp" LIB_EXTENTION
 
 /* Global variables */
 struct backends_api_functions {