Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
OpenDAS
vllm_cscc
Commits
c5602f0b
Unverified
Commit
c5602f0b
authored
Jun 10, 2024
by
Kevin H. Luu
Committed by
GitHub
Jun 10, 2024
Browse files
[ci] Mount buildkite agent on Docker container to upload benchmark results (#5330)
Signed-off-by:
kevin
<
kevin@anyscale.com
>
parent
f7f9c5f9
Changes
2
Show whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
6 additions
and
3 deletions
+6
-3
.buildkite/run-benchmarks.sh
.buildkite/run-benchmarks.sh
+3
-3
.buildkite/test-template-aws.j2
.buildkite/test-template-aws.j2
+3
-0
No files found.
.buildkite/run-benchmarks.sh
View file @
c5602f0b
...
...
@@ -54,12 +54,12 @@ tail -n 24 benchmark_serving.txt >> benchmark_results.md # last 24 lines
echo
'```'
>>
benchmark_results.md
# if the agent binary is not found, skip uploading the results, exit 0
if
[
!
-f
/workspace/
buildkite-agent
]
;
then
if
[
!
-f
buildkite-agent
]
;
then
exit
0
fi
# upload the results to buildkite
/workspace/
buildkite-agent annotate
--style
"info"
--context
"benchmark-results"
< benchmark_results.md
buildkite-agent annotate
--style
"info"
--context
"benchmark-results"
< benchmark_results.md
# exit with the exit code of the benchmarks
if
[
$bench_latency_exit_code
-ne
0
]
;
then
...
...
@@ -75,4 +75,4 @@ if [ $bench_serving_exit_code -ne 0 ]; then
fi
rm
ShareGPT_V3_unfiltered_cleaned_split.json
/workspace/
buildkite-agent artifact upload
"*.json"
buildkite-agent artifact upload
"*.json"
.buildkite/test-template-aws.j2
View file @
c5602f0b
...
...
@@ -49,6 +49,9 @@ steps:
{% if not step.no_gpu %}
gpus: all
{% endif %}
{% if step.label == "Benchmarks" %}
mount-buildkite-agent: true
{% endif %}
command: ["bash", "-c", "cd {{ (step.working_dir or default_working_dir) | safe }} && {{ step.command or (step.commands | join(' && ')) | safe }}"]
environment:
- VLLM_USAGE_SOURCE=ci-test
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment