Merge commit 'd803bfe2b1fe7f5e219e50ac20d6801a0a58ac75' as 'vendor/ruvector'

2026-02-28 14:39:40 -05:00
parent 7885bf6278 d803bfe2b1
commit cd5943df23
7854 changed files with 3522914 additions and 0 deletions
--- a/vendor/ruvector/crates/ruvector-postgres/benches/scripts/run_benchmarks.sh
+++ b/vendor/ruvector/crates/ruvector-postgres/benches/scripts/run_benchmarks.sh
@@ -0,0 +1,173 @@
+#!/bin/bash
+# Comprehensive benchmark runner script
+
+set -e
+
+# Colors for output
+RED='\033[0;31m'
+GREEN='\033[0;32m'
+YELLOW='\033[1;33m'
+BLUE='\033[0;34m'
+NC='\033[0m' # No Color
+
+# Configuration
+BENCHMARK_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")/.." && pwd)"
+RESULTS_DIR="${BENCHMARK_DIR}/results"
+TIMESTAMP=$(date +%Y%m%d_%H%M%S)
+
+# Create results directory
+mkdir -p "${RESULTS_DIR}"
+
+echo -e "${BLUE}==================================================${NC}"
+echo -e "${BLUE}  RuVector Comprehensive Benchmark Suite${NC}"
+echo -e "${BLUE}==================================================${NC}"
+echo ""
+
+# ============================================================================
+# Rust Benchmarks
+# ============================================================================
+
+echo -e "${GREEN}Running Rust benchmarks...${NC}"
+echo ""
+
+# Distance benchmarks
+echo -e "${YELLOW}1. Distance function benchmarks${NC}"
+cargo bench --bench distance_bench -- --output-format bencher | tee "${RESULTS_DIR}/distance_${TIMESTAMP}.txt"
+
+# Index benchmarks
+echo -e "${YELLOW}2. HNSW index benchmarks${NC}"
+cargo bench --bench index_bench -- --output-format bencher | tee "${RESULTS_DIR}/index_${TIMESTAMP}.txt"
+
+# Quantization benchmarks
+echo -e "${YELLOW}3. Quantization benchmarks${NC}"
+cargo bench --bench quantization_bench -- --output-format bencher | tee "${RESULTS_DIR}/quantization_${TIMESTAMP}.txt"
+
+# Quantized distance benchmarks
+echo -e "${YELLOW}4. Quantized distance benchmarks${NC}"
+cargo bench --bench quantized_distance_bench -- --output-format bencher | tee "${RESULTS_DIR}/quantized_distance_${TIMESTAMP}.txt"
+
+# ============================================================================
+# SQL Benchmarks (if PostgreSQL is available)
+# ============================================================================
+
+if command -v psql &> /dev/null; then
+    echo ""
+    echo -e "${GREEN}Running SQL benchmarks...${NC}"
+    echo ""
+
+    # Check if test database exists
+    if psql -lqt | cut -d \| -f 1 | grep -qw ruvector_bench; then
+        echo -e "${YELLOW}5. Quick SQL benchmark${NC}"
+        psql -d ruvector_bench -f "${BENCHMARK_DIR}/sql/quick_benchmark.sql" | tee "${RESULTS_DIR}/sql_quick_${TIMESTAMP}.txt"
+
+        echo -e "${YELLOW}6. Full workload benchmark${NC}"
+        echo -e "${RED}Warning: This may take several minutes...${NC}"
+        psql -d ruvector_bench -f "${BENCHMARK_DIR}/sql/benchmark_workload.sql" | tee "${RESULTS_DIR}/sql_workload_${TIMESTAMP}.txt"
+    else
+        echo -e "${YELLOW}Skipping SQL benchmarks (database 'ruvector_bench' not found)${NC}"
+        echo -e "${YELLOW}To run SQL benchmarks:${NC}"
+        echo -e "  createdb ruvector_bench"
+        echo -e "  psql -d ruvector_bench -c 'CREATE EXTENSION ruvector;'"
+        echo -e "  psql -d ruvector_bench -c 'CREATE EXTENSION pgvector;'"
+    fi
+else
+    echo -e "${YELLOW}Skipping SQL benchmarks (psql not found)${NC}"
+fi
+
+# ============================================================================
+# Generate Summary Report
+# ============================================================================
+
+echo ""
+echo -e "${GREEN}Generating summary report...${NC}"
+
+cat > "${RESULTS_DIR}/summary_${TIMESTAMP}.md" <<EOF
+# RuVector Benchmark Results
+
+**Date:** $(date)
+**Platform:** $(uname -s) $(uname -m)
+**Rust Version:** $(rustc --version)
+
+## Benchmark Files
+
+- Distance functions: \`distance_${TIMESTAMP}.txt\`
+- HNSW index: \`index_${TIMESTAMP}.txt\`
+- Quantization: \`quantization_${TIMESTAMP}.txt\`
+- Quantized distance: \`quantized_distance_${TIMESTAMP}.txt\`
+
+## SQL Benchmarks
+
+EOF
+
+if [ -f "${RESULTS_DIR}/sql_quick_${TIMESTAMP}.txt" ]; then
+    cat >> "${RESULTS_DIR}/summary_${TIMESTAMP}.md" <<EOF
+- Quick benchmark: \`sql_quick_${TIMESTAMP}.txt\`
+- Full workload: \`sql_workload_${TIMESTAMP}.txt\`
+
+EOF
+else
+    cat >> "${RESULTS_DIR}/summary_${TIMESTAMP}.md" <<EOF
+SQL benchmarks were not run. See setup instructions above.
+
+EOF
+fi
+
+cat >> "${RESULTS_DIR}/summary_${TIMESTAMP}.md" <<EOF
+## System Information
+
+\`\`\`
+$(uname -a)
+\`\`\`
+
+### CPU Information
+
+\`\`\`
+$(lscpu 2>/dev/null || sysctl -a | grep machdep.cpu || echo "CPU info not available")
+\`\`\`
+
+### Memory Information
+
+\`\`\`
+$(free -h 2>/dev/null || vm_stat || echo "Memory info not available")
+\`\`\`
+
+## Running the Benchmarks
+
+To reproduce these results:
+
+\`\`\`bash
+cd crates/ruvector-postgres
+bash benches/scripts/run_benchmarks.sh
+\`\`\`
+
+## Comparing with Previous Results
+
+\`\`\`bash
+# Install cargo-criterion for better comparison
+cargo install cargo-criterion
+
+# Run with baseline
+cargo criterion --bench distance_bench --baseline main
+\`\`\`
+EOF
+
+echo ""
+echo -e "${GREEN}==================================================${NC}"
+echo -e "${GREEN}  Benchmark Complete!${NC}"
+echo -e "${GREEN}==================================================${NC}"
+echo ""
+echo -e "Results saved to: ${BLUE}${RESULTS_DIR}${NC}"
+echo -e "Summary report: ${BLUE}${RESULTS_DIR}/summary_${TIMESTAMP}.md${NC}"
+echo ""
+
+# ============================================================================
+# Optional: Open results in browser if criterion HTML is available
+# ============================================================================
+
+if [ -d "target/criterion" ]; then
+    echo -e "${YELLOW}Criterion HTML reports available at:${NC}"
+    echo -e "  ${BLUE}file://$(pwd)/target/criterion/report/index.html${NC}"
+fi
+
+echo ""
+echo -e "${GREEN}Done!${NC}"