ci: decouple stage and runner for cuda registry (#25197)
This commit is contained in:
@@ -30,7 +30,7 @@ CALIBRATION_BATCH_SIZE = 36
|
||||
CALIBRATION_NUM_SAMPLES = 512
|
||||
DEFAULT_DEVICE = "cuda:0"
|
||||
|
||||
register_cuda_ci(est_time=11, suite="stage-b-test-1-gpu-small")
|
||||
register_cuda_ci(est_time=11, stage="stage-b", runner_config="1-gpu-small")
|
||||
|
||||
|
||||
class TestModelOptModelLoader(CustomTestCase):
|
||||
|
||||
Reference in New Issue
Block a user