Skip to content
This repository was archived by the owner on Oct 11, 2024. It is now read-only.

Commit a9bd95b

Browse files
khluuRobert Shaw
authored andcommitted
[ci] Mount buildkite agent on Docker container to upload benchmark results (vllm-project#5330)
Signed-off-by: kevin <kevin@anyscale.com>
1 parent 9d82433 commit a9bd95b

File tree

2 files changed

+6
-3
lines changed

2 files changed

+6
-3
lines changed

.buildkite/run-benchmarks.sh

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -54,12 +54,12 @@ tail -n 24 benchmark_serving.txt >> benchmark_results.md # last 24 lines
5454
echo '```' >> benchmark_results.md
5555

5656
# if the agent binary is not found, skip uploading the results, exit 0
57-
if [ ! -f /workspace/buildkite-agent ]; then
57+
if [ ! -f buildkite-agent ]; then
5858
exit 0
5959
fi
6060

6161
# upload the results to buildkite
62-
/workspace/buildkite-agent annotate --style "info" --context "benchmark-results" < benchmark_results.md
62+
buildkite-agent annotate --style "info" --context "benchmark-results" < benchmark_results.md
6363

6464
# exit with the exit code of the benchmarks
6565
if [ $bench_latency_exit_code -ne 0 ]; then
@@ -75,4 +75,4 @@ if [ $bench_serving_exit_code -ne 0 ]; then
7575
fi
7676

7777
rm ShareGPT_V3_unfiltered_cleaned_split.json
78-
/workspace/buildkite-agent artifact upload "*.json"
78+
buildkite-agent artifact upload "*.json"

.buildkite/test-template-aws.j2

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -49,6 +49,9 @@ steps:
4949
{% if not step.no_gpu %}
5050
gpus: all
5151
{% endif %}
52+
{% if step.label == "Benchmarks" %}
53+
mount-buildkite-agent: true
54+
{% endif %}
5255
command: ["bash", "-c", "cd {{ (step.working_dir or default_working_dir) | safe }} && {{ step.command or (step.commands | join(' && ')) | safe }}"]
5356
environment:
5457
- VLLM_USAGE_SOURCE=ci-test

0 commit comments

Comments
 (0)