fix compile issue on ARM

spider-gazelle · Jan 7, 2023 · 8aec56b · 8aec56b
1 parent 4e72c6f
commit 8aec56b
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 5 deletions.
diff --git a/build_tensorflowlite.sh b/build_tensorflowlite.sh
@@ -22,21 +22,21 @@ echo "--"
 
 mkdir tflite_build
 cd tflite_build
-cmake ../tensorflow/tensorflow/lite/c
+cmake ../tensorflow/tensorflow/lite/c -DTFLITE_ENABLE_GPU=ON
 
 echo "--"
 echo "building..."
 echo "--"
 
-cmake --build . -j
+cmake --build . -j2 || true
 
 FILE=./libtensorflowlite_c.so
 if test -f "$FILE"; then
   echo "--"
   echo "build success!"
 else
   echo "build failed, retrying..."
-  cmake --build . -j
+  cmake --build . -j1
 
   if test -f "$FILE"; then
     echo "--"

diff --git a/shard.yml b/shard.yml
@@ -1,5 +1,5 @@
 name: tensorflow_lite
-version: 0.9.1
+version: 0.9.2
 
 development_dependencies:
   ameba:

diff --git a/src/tensorflow_lite/lib_tensorflowlite.cr b/src/tensorflow_lite/lib_tensorflowlite.cr
@@ -204,7 +204,7 @@ lib LibTensorflowLite
   type Delegate = Void*
   fun interpreter_options_add_opaque_delegate = TfLiteInterpreterOptionsAddOpaqueDelegate(options : InterpreterOptions, opaque_delegate : OpaqueDelegateStruct)
   type OpaqueDelegateStruct = Void*
-  fun interpreter_options_set_error_reporter = TfLiteInterpreterOptionsSetErrorReporter(options : InterpreterOptions, reporter : (Void*, LibC::Char*, VaList -> Void), user_data : Void*)
+  fun interpreter_options_set_error_reporter = TfLiteInterpreterOptionsSetErrorReporter(options : InterpreterOptions, reporter : (Void*, LibC::Char*, Void* -> Void), user_data : Void*)
   fun interpreter_options_add_registration_external = TfLiteInterpreterOptionsAddRegistrationExternal(options : InterpreterOptions, registration : RegistrationExternal)
   type RegistrationExternal = Void*
   fun interpreter_options_enable_cancellation = TfLiteInterpreterOptionsEnableCancellation(options : InterpreterOptions, enable : LibC::Int) : Status