feat: fp16 vector

.cargo/config.toml

@@ -4,3 +4,13 @@ rustdocflags = ["--document-private-items"]
 [target.'cfg(target_os="macos")']
 # Postgres symbols won't be available until runtime
 rustflags = ["-Clink-arg=-Wl,-undefined,dynamic_lookup"]
+
+[target.x86_64-unknown-linux-gnu]
+linker = "x86_64-linux-gnu-gcc"
+
+[target.aarch64-unknown-linux-gnu]
+linker = "aarch64-linux-gnu-gcc"
+
+[env]
+BINDGEN_EXTRA_CLANG_ARGS_x86_64_unknown_linux_gnu = "-isystem /usr/x86_64-linux-gnu/include/ -ccc-gcc-name x86_64-linux-gnu-gcc"
+BINDGEN_EXTRA_CLANG_ARGS_aarch64_unknown_linux_gnu = "-isystem /usr/aarch64-linux-gnu/include/ -ccc-gcc-name aarch64-linux-gnu-gcc"

.github/workflows/check.yml

@@ -6,6 +6,7 @@ on:
     paths:
       - ".cargo/**"
       - ".github/**"
+      - "crates/**"
       - "scripts/**"
       - "src/**"
       - "tests/**"
@@ -18,6 +19,7 @@ on:
     paths:
       - ".cargo/**"
       - ".github/**"
+      - "crates/**"
       - "scripts/**"
       - "src/**"
       - "tests/**"
@@ -90,11 +92,16 @@ jobs:
       - name: Format check
         run: cargo fmt --check
       - name: Semantic check
-        run: cargo clippy --no-default-features --features "pg${{ matrix.version }} pg_test"
+        run: |
+          cargo clippy --no-default-features --features "pg${{ matrix.version }} pg_test" --target x86_64-unknown-linux-gnu
+          cargo clippy --no-default-features --features "pg${{ matrix.version }} pg_test" --target aarch64-unknown-linux-gnu
       - name: Debug build
-        run: cargo build --no-default-features --features "pg${{ matrix.version }} pg_test"
+        run: |
+          cargo build --no-default-features --features "pg${{ matrix.version }} pg_test" --target x86_64-unknown-linux-gnu
+          cargo build --no-default-features --features "pg${{ matrix.version }} pg_test" --target aarch64-unknown-linux-gnu
       - name: Test
-        run: cargo test --all --no-default-features --features "pg${{ matrix.version }} pg_test" -- --nocapture
+        run: |
+          cargo test --all --no-fail-fast --no-default-features --features "pg${{ matrix.version }} pg_test" --target x86_64-unknown-linux-gnu
       - name: Install release
         run: ./scripts/ci_install.sh
       - name: Sqllogictest

.github/workflows/release.yml

@@ -112,15 +112,17 @@ jobs:
       - uses: mozilla-actions/[email protected]
       - name: Prepare
         run: |
-          sudo sh -c 'echo "deb http://apt.postgresql.org/pub/repos/apt $(lsb_release -cs)-pgdg main" > /etc/apt/sources.list.d/pgdg.list'
+          sudo sh -c 'echo "deb http://apt.postgresql.org/pub/repos/apt $(lsb_release -cs)-pgdg main" >> /etc/apt/sources.list.d/pgdg.list'
+          sudo sh -c 'echo "deb http://apt.llvm.org/$(lsb_release -cs)/ llvm-toolchain-$(lsb_release -cs)-16 main" >> /etc/apt/sources.list'
           wget --quiet -O - https://www.postgresql.org/media/keys/ACCC4CF8.asc | sudo apt-key add -
+          wget --quiet -O - https://apt.llvm.org/llvm-snapshot.gpg.key | sudo apt-key add -
           sudo apt-get update
           sudo apt-get -y install libpq-dev postgresql-${{ matrix.version }} postgresql-server-dev-${{ matrix.version }}
+          sudo apt-get -y install clang-16
           cargo install cargo-pgrx --git https://github.com/tensorchord/pgrx.git --rev $(cat Cargo.toml | grep "pgrx =" | awk -F'rev = "' '{print $2}' | cut -d'"' -f1)
           cargo pgrx init --pg${{ matrix.version }}=/usr/lib/postgresql/${{ matrix.version }}/bin/pg_config
           if [[ "${{ matrix.arch }}" == "arm64" ]]; then
             sudo apt-get -y install crossbuild-essential-arm64
-            rustup target add aarch64-unknown-linux-gnu
           fi
       - name: Build Release
         id: build_release
@@ -130,8 +132,6 @@ jobs:
           mkdir ./artifacts
           cargo pgrx package
           if [[ "${{ matrix.arch }}" == "arm64" ]]; then
-            export CARGO_TARGET_AARCH64_UNKNOWN_LINUX_GNU_LINKER=aarch64-linux-gnu-gcc
-            export BINDGEN_EXTRA_CLANG_ARGS_aarch64_unknown_linux_gnu="-target aarch64-unknown-linux-gnu -isystem /usr/aarch64-linux-gnu/include/ -ccc-gcc-name aarch64-linux-gnu-gcc"
             cargo build --target aarch64-unknown-linux-gnu --release --features "pg${{ matrix.version }}" --no-default-features
             mv ./target/aarch64-unknown-linux-gnu/release/libvectors.so ./target/release/vectors-pg${{ matrix.version }}/usr/lib/postgresql/${{ matrix.version }}/lib/vectors.so
           fi

.gitignore

@@ -6,4 +6,5 @@
 .vscode
 .ignore
 __pycache__
-.pytest_cache
+.pytest_cache
+rustc-ice-*.txt