diff --git a/.github/workflows/benchmark-local.yaml b/.github/workflows/performance-comparisons.yaml similarity index 84% rename from .github/workflows/benchmark-local.yaml rename to .github/workflows/performance-comparisons.yaml index d514a33522..f243b7b6e4 100644 --- a/.github/workflows/benchmark-local.yaml +++ b/.github/workflows/performance-comparisons.yaml @@ -54,5 +54,14 @@ jobs: ray up .github/assets/ray.yaml -y HEAD_NODE_IP=$(ray get-head-ip .github/assets/ray.yaml | tail -n 1) ssh -o StrictHostKeyChecking=no -fN -L 8265:localhost:8265 -i ~/.ssh/ci-github-actions-ray-cluster-key.pem ubuntu@$HEAD_NODE_IP + DAFT_RUNNER=ray python -m benchmarking.tpch \ + --skip_questions="2,3,4,5,6,7,8,9,10,11,12,13,14,15,16,17,18,19,20,21,22" \ + --num_parts 2 \ + --scale_factor 2 \ + --parquet_file_cache /tmp/data \ + --output_csv output.csv \ + --ray_job_dashboard_url http://localhost:8265 \ + --skip_warmup + ray job submit --working-dir .github/working_dir --address http://localhost:8265 -- python test.py ray down .github/assets/ray.yaml -y