Continuous Benchmark 2 #163
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
name: Continuous Benchmark 2 | |
on: | |
workflow_dispatch: | |
inputs: | |
feature_flags_all: | |
type: boolean | |
description: "Enable all feature flags (true by default)" | |
default: true | |
schedule: | |
# Run every day at midnight | |
- cron: "0 0 * * *" | |
# Restrict to only running this workflow one at a time. | |
# Any new runs will be queued until the previous run is complete. | |
# Any existing pending runs will be cancelled and replaced with current run. | |
concurrency: | |
group: continuous-benchmark | |
jobs: | |
# Schedule this benchmark to run once a day for the sake of saving on S3 costs. | |
runLoadTimeBenchmark: | |
runs-on: ubuntu-latest | |
steps: | |
- uses: actions/checkout@v3 | |
- uses: webfactory/[email protected] | |
with: | |
ssh-private-key: ${{ secrets.SSH_PRIVATE_KEY }} | |
- name: Benches | |
id: benches | |
run: | | |
export HCLOUD_TOKEN=${{ secrets.HCLOUD_TOKEN }} | |
export POSTGRES_PASSWORD=${{ secrets.POSTGRES_PASSWORD }} | |
export POSTGRES_HOST=${{ secrets.POSTGRES_HOST }} | |
export SERVER_NAME="benchmark-server-3" | |
export QDRANT__FEATURE_FLAGS__ALL=${{ inputs.feature_flags_all }} | |
bash -x tools/setup_ci.sh | |
set +e | |
# Benchmark collection load time | |
export BENCHMARK_STRATEGY="collection-reload" | |
declare -A DATASET_TO_ENGINE | |
declare -A DATASET_TO_URL | |
DATASET_TO_ENGINE["all-payloads-default"]="qdrant-continuous-benchmark-snapshot" | |
DATASET_TO_ENGINE["all-payloads-on-disk"]="qdrant-continuous-benchmark-snapshot" | |
DATASET_TO_ENGINE["all-payloads-default-sparse"]="qdrant-continuous-benchmark-snapshot" | |
DATASET_TO_ENGINE["all-payloads-on-disk-sparse"]="qdrant-continuous-benchmark-snapshot" | |
export STORAGE_URL="https://storage.googleapis.com/qdrant-benchmark-snapshots/all-payloads" | |
DATASET_TO_URL["all-payloads-default"]="${STORAGE_URL}/benchmark-all-payloads-500k-768-default.snapshot" | |
DATASET_TO_URL["all-payloads-on-disk"]="${STORAGE_URL}/benchmark-all-payloads-500k-768-on-disk.snapshot" | |
DATASET_TO_URL["all-payloads-default-sparse"]="${STORAGE_URL}/benchmark-all-payloads-500k-sparse-default.snapshot" | |
DATASET_TO_URL["all-payloads-on-disk-sparse"]="${STORAGE_URL}/benchmark-all-payloads-500k-sparse-on-disk.snapshot" | |
set +e | |
for dataset in "${!DATASET_TO_ENGINE[@]}"; do | |
export ENGINE_NAME=${DATASET_TO_ENGINE[$dataset]} | |
export DATASETS=$dataset | |
export SNAPSHOT_URL=${DATASET_TO_URL[$dataset]} | |
# Benchmark the dev branch: | |
export QDRANT_VERSION=ghcr/dev | |
export QDRANT__FEATURE_FLAGS__ALL=true | |
timeout 30m bash -x tools/run_ci.sh | |
# Benchmark the master branch: | |
export QDRANT_VERSION=docker/master | |
export QDRANT__FEATURE_FLAGS__ALL=false | |
timeout 30m bash -x tools/run_ci.sh | |
done | |
set -e | |
- name: Fail job if any of the benches failed | |
if: steps.benches.outputs.failed == 'error' || steps.benches.outputs.failed == 'timeout' | |
run: exit 1 | |
- name: Send slack message | |
uses: ./.github/workflows/actions/send-slack-msg | |
if: failure() || cancelled() | |
with: | |
bench_name: "runLoadTimeBenchmark" | |
job_status: ${{ job.status }} | |
failed_outputs: ${{ steps.benches.outputs.failed }} | |
qdrant_version: ${{ steps.benches.outputs.qdrant_version }} | |
engine_name: ${{ steps.benches.outputs.engine_name }} | |
dataset: ${{ steps.benches.outputs.dataset }} | |
env: | |
SLACK_WEBHOOK_URL: ${{ secrets.CI_ALERTS_CHANNEL_WEBHOOK_URL }} | |
SLACK_WEBHOOK_TYPE: INCOMING_WEBHOOK |