[ci] Switch some CI jobs to H200 MIG slices (#38956)

2026-04-05 13:26:11 -07:00
parent 4dd49b06f8
commit 56de443db1
14 changed files with 25 additions and 0 deletions
--- a/.buildkite/test_areas/models_language.yaml
+++ b/.buildkite/test_areas/models_language.yaml
@@ -67,6 +67,7 @@ steps:

 - label: Language Models Test (PPL)
  timeout_in_minutes: 110
+  device: h200_18gb
  optional: true
  source_file_dependencies:
  - vllm/
@@ -90,6 +91,7 @@ steps:

 - label: Language Models Test (MTEB)
  timeout_in_minutes: 110
+  device: h200_18gb
  optional: true
  source_file_dependencies:
  - vllm/