Skip to content

Commit e141125

Browse files
update benchmark directory for vllm (#84)
Signed-off-by: Huy Do <huydhn@gmail.com> Co-authored-by: Huy Do <huydhn@gmail.com>
1 parent e891bae commit e141125

File tree

2 files changed

+6
-6
lines changed

2 files changed

+6
-6
lines changed

.github/scripts/setup_vllm_benchmark.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -50,7 +50,7 @@ def parse_args() -> Any:
5050
parser.add_argument(
5151
"--to-benchmark-configs-dir",
5252
type=str,
53-
default=".buildkite/nightly-benchmarks/tests",
53+
default=".buildkite/performance-benchmarks/tests",
5454
action=ValidateDir,
5555
help="a subset of vLLM benchmark configs to run on this runner",
5656
required=True,

.github/workflows/vllm-benchmark.yml

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -239,19 +239,19 @@ jobs:
239239
240240
pushd vllm-benchmarks/vllm
241241
git checkout "${HEAD_SHA}"
242-
rm .buildkite/nightly-benchmarks/tests/*.json
242+
rm .buildkite/performance-benchmarks/tests/*.json
243243
popd
244244
245245
# Set the list of benchmarks we want to cover in this runner
246246
python3 .github/scripts/setup_vllm_benchmark.py \
247247
--from-benchmark-configs-dir vllm-benchmarks/benchmarks \
248-
--to-benchmark-configs-dir vllm-benchmarks/vllm/.buildkite/nightly-benchmarks/tests \
248+
--to-benchmark-configs-dir vllm-benchmarks/vllm/.buildkite/performance-benchmarks/tests \
249249
--models "${MODELS}" \
250250
--device "${DEVICE_NAME}"
251251
252252
pushd vllm-benchmarks/vllm
253-
ls -lah .buildkite/nightly-benchmarks/tests
254-
find .buildkite/nightly-benchmarks/tests -type f -exec cat {} \;
253+
ls -lah .buildkite/performance-benchmarks/tests
254+
find .buildkite/performance-benchmarks/tests -type f -exec cat {} \;
255255
popd
256256
257257
- name: Run vLLM benchmark
@@ -292,7 +292,7 @@ jobs:
292292
-w /tmp/workspace \
293293
"${DOCKER_IMAGE}"
294294
)
295-
docker exec -t "${container_name}" bash -c "cd vllm-benchmarks/vllm && bash .buildkite/nightly-benchmarks/scripts/run-performance-benchmarks.sh"
295+
docker exec -t "${container_name}" bash -c "cd vllm-benchmarks/vllm && bash .buildkite/performance-benchmarks/scripts/run-performance-benchmarks.sh"
296296
297297
- name: Authenticate with AWS
298298
# AWS CUDA runners already have access to the bucket via its runner IAM role

0 commit comments

Comments
 (0)