adapt openGauss

Author: createart

ann_benchmarks/algorithms/openGauss/Dockerfile

@@ -0,0 +1,39 @@
+FROM ann-benchmarks
+
+# https://github.com/pgvector/pgvector/blob/master/Dockerfile
+
+RUN git clone https://github.com/pgvector/pgvector /tmp/pgvector
+
+RUN DEBIAN_FRONTEND=noninteractive apt-get -y install tzdata
+RUN apt-get update && apt-get install -y --no-install-recommends build-essential postgresql-common
+RUN /usr/share/postgresql-common/pgdg/apt.postgresql.org.sh -y
+RUN apt-get install -y --no-install-recommends postgresql-16 postgresql-server-dev-16
+RUN sh -c 'echo "local all all trust" > /etc/postgresql/16/main/pg_hba.conf'
+
+# Dynamically set OPTFLAGS based on the architecture
+RUN ARCH=$(uname -m) && \
+    if [ "$ARCH" = "aarch64" ]; then \
+        OPTFLAGS="-march=native -msve-vector-bits=512"; \
+    elif [ "$ARCH" = "x86_64" ]; then \
+        OPTFLAGS="-march=native -mprefer-vector-width=512"; \
+    else \
+        OPTFLAGS="-march=native"; \
+    fi && \
+    cd /tmp/pgvector && \
+    make clean && \
+    make OPTFLAGS="$OPTFLAGS" && \
+    make install
+
+USER postgres
+RUN service postgresql start && \
+    psql -c "CREATE USER ann WITH ENCRYPTED PASSWORD 'ann'" && \
+    psql -c "CREATE DATABASE ann" && \
+    psql -c "GRANT ALL PRIVILEGES ON DATABASE ann TO ann" && \
+    psql -d ann -c "GRANT ALL ON SCHEMA public TO ann" && \
+    psql -d ann -c "CREATE EXTENSION vector" && \
+    psql -c "ALTER USER ann SET maintenance_work_mem = '4GB'" && \
+    psql -c "ALTER USER ann SET max_parallel_maintenance_workers = 0" && \
+    psql -c "ALTER SYSTEM SET shared_buffers = '4GB'"
+USER root
+
+RUN pip install psycopg[binary] pgvector

ann_benchmarks/algorithms/openGauss/config.yml

@@ -0,0 +1,42 @@
+float:
+  any:
+  - base_args: ['@metric']
+    constructor: openGaussHNSWPQ
+    disabled: false
+    docker_tag: ann-benchmarks-openGauss
+    module: ann_benchmarks.algorithms.openGauss
+    name: openGauss-hnswpq
+    run_groups:
+      M-16:
+        arg_groups: [{M: 16, efConstruction: 200, hnswEarlystopThreshold: 2147483647, pqM: 96, concurrents: 80}]
+        args: {}
+        query_args: [[10, 20, 40, 80, 120, 200, 400, 800]]
+      M-24:
+        arg_groups: [{M: 24, efConstruction: 200, hnswEarlystopThreshold: 2147483647, pqM: 96, concurrents: 80}]
+        args: {}
+        query_args: [[10, 20, 40, 80, 120, 200, 400, 800]]
+  - base_args: ['@metric']
+    constructor: openGaussHNSW
+    disabled: false
+    docker_tag: ann-benchmarks-openGauss
+    module: ann_benchmarks.algorithms.openGauss
+    name: openGauss-hnsw
+    run_groups:
+      M-16:
+        arg_groups: [{M: 16, efConstruction: 200, concurrents: 80}]
+        args: {}
+        query_args: [[10, 20, 40, 80, 120, 200, 400, 800]]
+      M-24:
+        arg_groups: [{M: 24, efConstruction: 200, concurrents: 80}]
+        args: {}
+        query_args: [[10, 20, 40, 80, 120, 200, 400, 800]]
+  - base_args: ['@metric']
+    constructor: openGaussIVF
+    disabled: false
+    docker_tag: ann-benchmarks-openGauss
+    module: ann_benchmarks.algorithms.openGauss
+    name: openGauss-ivfflat
+    run_groups:
+      pgvector:
+        args: [[100, 200, 400, 1000, 2000, 4000]]
+        query_args: [[1, 2, 4, 10, 20, 40, 100]]