Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion bench/bench-memory.sh
Original file line number Diff line number Diff line change
Expand Up @@ -65,7 +65,7 @@ fi

get_redis_memory() {
local port=$1
redis-cli -p "$port" info memory 2>/dev/null | grep used_memory: | cut -d: -f2 | tr -d '\r'
redis-cli -p "$port" info memory 2>/dev/null | grep "^used_memory:" | cut -d: -f2 | tr -d '\r'
}

wait_for_server() {
Expand Down
2 changes: 1 addition & 1 deletion bench/bench-proto.sh
Original file line number Diff line number Diff line change
Expand Up @@ -90,7 +90,7 @@ echo "port: $EMBER_PORT"
echo ""

echo "starting ember (with protobuf support) on port $EMBER_PORT..."
"$EMBER_BIN" --port "$EMBER_PORT" > /dev/null 2>&1 &
"$EMBER_BIN" --port "$EMBER_PORT" --protobuf > /dev/null 2>&1 &
EMBER_PID=$!

retries=50
Expand Down
41 changes: 38 additions & 3 deletions bench/setup-vm.sh
Original file line number Diff line number Diff line change
Expand Up @@ -11,8 +11,28 @@ set -e
echo "=== installing system dependencies ==="

sudo apt-get update
sudo apt-get install -y build-essential git redis-server \
autoconf automake libpcre3-dev libevent-dev pkg-config zlib1g-dev libssl-dev
sudo apt-get install -y build-essential git redis-server unzip \
autoconf automake libpcre3-dev libevent-dev pkg-config zlib1g-dev libssl-dev \
gcc-12 g++-12 python3-venv

# prefer gcc-12 (needed for usearch SIMD in vector feature)
if command -v gcc-12 &> /dev/null; then
sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-12 120 \
--slave /usr/bin/g++ g++ /usr/bin/g++-12 2>/dev/null || true
fi

# disable the redis systemd service — benchmarks manage their own redis instances
sudo systemctl stop redis-server 2>/dev/null || true
sudo systemctl disable redis-server 2>/dev/null || true

# install modern protoc (needed for gRPC and protobuf features)
if ! protoc --version 2>/dev/null | grep -q "29"; then
echo "installing protoc 29.3..."
curl -fsSL "https://github.com/protocolbuffers/protobuf/releases/download/v29.3/protoc-29.3-linux-x86_64.zip" -o /tmp/protoc.zip
sudo unzip -o /tmp/protoc.zip -d /usr/local
rm -f /tmp/protoc.zip
protoc --version
fi

echo ""
echo "=== installing rust ==="
Expand Down Expand Up @@ -94,7 +114,22 @@ fi
echo ""
echo "=== building ember ==="

cargo build --release --features jemalloc
# detect available features
FEATURES="jemalloc"
if grep -q 'vector' crates/ember-server/Cargo.toml 2>/dev/null; then
FEATURES="$FEATURES,vector"
fi
if grep -q 'grpc' crates/ember-server/Cargo.toml 2>/dev/null; then
FEATURES="$FEATURES,grpc"
fi
if grep -q 'protobuf' crates/ember-server/Cargo.toml 2>/dev/null; then
FEATURES="$FEATURES,protobuf"
fi
if grep -q 'encryption' crates/ember-server/Cargo.toml 2>/dev/null; then
FEATURES="$FEATURES,encryption"
fi
echo "building with features: $FEATURES"
cargo build --release -p ember-server --features "$FEATURES"

echo ""
echo "=== verifying ==="
Expand Down