Unverified Commit 8bdd3979 authored by Kevin H. Luu's avatar Kevin H. Luu Committed by GitHub
Browse files

[CI] Change GPU key to device key for B200 test (#33275)


Signed-off-by: default avatarkhluu <khluu000@gmail.com>
parent c4e744db
...@@ -61,7 +61,7 @@ steps: ...@@ -61,7 +61,7 @@ steps:
- pytest -s -v evals/gsm8k/test_gsm8k_correctness.py --config-list-file=evals/gsm8k/configs/moe-refactor/config-h100.txt - pytest -s -v evals/gsm8k/test_gsm8k_correctness.py --config-list-file=evals/gsm8k/configs/moe-refactor/config-h100.txt
- label: MoE Refactor Integration Test (B200 - TEMPORARY) - label: MoE Refactor Integration Test (B200 - TEMPORARY)
gpu: b200 device: b200
optional: true optional: true
num_devices: 2 num_devices: 2
commands: commands:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment