From 4293c00b84b968ed25f80dfd2af3bb34d1eeeef6 Mon Sep 17 00:00:00 2001 From: Matthew Bonanni Date: Tue, 10 Feb 2026 16:04:07 -0500 Subject: [PATCH] [Benchmarks] Fix attention benchmark smoke test (#34269) Signed-off-by: Matthew Bonanni --- .buildkite/test_areas/benchmarks.yaml | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/.buildkite/test_areas/benchmarks.yaml b/.buildkite/test_areas/benchmarks.yaml index 57080c46f..a30ec60ea 100644 --- a/.buildkite/test_areas/benchmarks.yaml +++ b/.buildkite/test_areas/benchmarks.yaml @@ -22,9 +22,10 @@ steps: device: b200 num_gpus: 2 optional: true + working_dir: "/vllm-workspace/" timeout_in_minutes: 10 source_file_dependencies: - benchmarks/attention_benchmarks/ - vllm/v1/attention/ commands: - - python benchmarks/attention_benchmarks/benchmark.py --backends flash flashinfer --batch-specs "8q1s1k" --repeats 1 --warmup-iters 1 + - python3 benchmarks/attention_benchmarks/benchmark.py --backends flash flashinfer --batch-specs "8q1s1k" --repeats 1 --warmup-iters 1