Explorar el Código

fea(plugin): add arm64 labelImage plugin

Signed-off-by: Jianxiang Ran <jianxiang.ran@emqx.io>
Jianxiang Ran hace 2 años
padre
commit
86699149ed

+ 0 - 2
.github/workflows/build_packages.yaml

@@ -200,8 +200,6 @@ jobs:
         golang:
           - 1.18.5
         exclude:
-          - arch: linux/arm64
-            plugin: functions/labelImage
           - os: [alpine,"alpine"]
             plugin: functions/labelImage
           - os: [alpine,"alpine"]

+ 6 - 0
build-plugins.sh

@@ -45,6 +45,12 @@ build(){
             ;;
         labelImage )
             git clone -b v2.2.0-rc3 --depth 1 https://github.com/tensorflow/tensorflow.git /tmp/tensorflow;
+            if [ "$(uname -m)" = "x86_64" ]; then
+                mv $(pwd)/extensions/functions/labelImage/dependencies/amd64/*.so $(pwd)/extensions/functions/labelImage/lib
+            fi;
+            if [ "$(uname -m)" = "aarch64" ]; then
+                mv $(pwd)/extensions/functions/labelImage/dependencies/arm64/*.so $(pwd)/extensions/functions/labelImage/lib
+            fi;
             CGO_CFLAGS=-I/tmp/tensorflow CGO_LDFLAGS=-L$(pwd)/extensions/functions/labelImage/lib go build -trimpath -modfile extensions.mod --buildmode=plugin -o extensions/functions/labelImage/labelImage.so extensions/functions/labelImage/*.go
             ;;
         * )

extensions/functions/labelImage/lib/libtensorflowlite.so → extensions/functions/labelImage/dependencies/amd64/libtensorflowlite.so


extensions/functions/labelImage/lib/libtensorflowlite_c.so → extensions/functions/labelImage/dependencies/amd64/libtensorflowlite_c.so


BIN
extensions/functions/labelImage/dependencies/arm64/libtensorflowlite.so


BIN
extensions/functions/labelImage/dependencies/arm64/libtensorflowlite_c.so