pass usearch defines to enable simsimd and native f16 for optimized b…

…uilds
lanterndata · Sep 24, 2024 · b6df5fe · b6df5fe
1 parent c458731
commit b6df5fe
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 2 deletions.
diff --git a/.github/workflows/publish-cli-docker.yaml b/.github/workflows/publish-cli-docker.yaml
@@ -11,7 +11,7 @@ on:
         type: string
         description: "CLI version"
         required: true
-        default: "0.3.24"
+        default: "0.3.25"
       IMAGE_NAME:
         type: string
         description: "Container image name to tag"
@@ -90,6 +90,7 @@ jobs:
             linux/amd64
           build-args: |
             RUSTFLAGS=${{ format('-C target-cpu={0}', matrix.target) }}
+            CXXFLAGS=${{ format('-march={0} -DUSEARCH_USE_SIMSIMD=1 -DUSEARCH_USE_FP16LIB=0', matrix.target) }}
           file: Dockerfile.cli
           push: true
           tags: |

diff --git a/Dockerfile.cli b/Dockerfile.cli
@@ -5,6 +5,8 @@ COPY . .
 ENV ORT_STRATEGY="system"
 ARG RUSTFLAGS=""
 ENV RUSTFLAGS=$RUSTFLAGS
+ARG CXXFLAGS=""
+ENV CXXFLAGS=$CXXFLAGS
 
 # Remove lantern_extras from workspace
 RUN rm -rf lantern_extras && sed -i -e 's/"lantern_extras",//' Cargo.toml

diff --git a/lantern_cli/Cargo.toml b/lantern_cli/Cargo.toml
@@ -1,6 +1,6 @@
 [package]
 name = "lantern_cli"
-version = "0.3.24"
+version = "0.3.25"
 edition = "2021"
 
 [[bin]]