Add 5090 dry run stage to PR test workflow (#17022)

This commit is contained in:
Alison Shao
2026-01-13 14:12:33 -08:00
committed by GitHub
parent 339915ce2b
commit b880607108
71 changed files with 158 additions and 3 deletions
@@ -26,6 +26,7 @@ CALIBRATION_NUM_SAMPLES = 512
DEFAULT_DEVICE = "cuda:0"
register_cuda_ci(est_time=11, suite="stage-b-test-small-1-gpu")
register_cuda_ci(est_time=8, suite="stage-b-test-small-1-gpu-5090")
class TestModelOptModelLoader(CustomTestCase):