Skip to content

Commit 69e17e7

Browse files
authored
Add some coverage for A100 (#50)
We only have 1xA100 on OSS CI. Everything else has been moving to H100 Signed-off-by: Huy Do <huydhn@gmail.com>
1 parent 570572f commit 69e17e7

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

.github/scripts/generate_vllm_benchmark_matrix.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -14,6 +14,7 @@
1414
# while ROCm runner are provided by AMD
1515
TP_TO_RUNNER_MAPPING = {
1616
1: [
17+
"linux.aws.a100",
1718
"linux.aws.h100",
1819
"linux.rocm.gpu.mi300.2", # No single ROCm GPU?
1920
"linux.24xl.spr-metal",
@@ -38,6 +39,7 @@
3839

3940
# This mapping is needed to find out the platform of the runner
4041
RUNNER_TO_PLATFORM_MAPPING = {
42+
"linux.aws.a100": "cuda",
4143
"linux.aws.h100": "cuda",
4244
"linux.aws.h100.4": "cuda",
4345
"linux.aws.h100.8": "cuda",

0 commit comments

Comments
 (0)