experiment: serialize crossover launch locks

1 files changed, 9 insertions, 0 deletions

diff --git a/experiments/crossover_artifact_validation_smoke.py b/experiments/crossover_artifact_validation_smoke.py
index e267832..f410a3e 100644
--- a/experiments/crossover_artifact_validation_smoke.py
+++ b/experiments/crossover_artifact_validation_smoke.py
@@ -44,6 +44,15 @@ def main():
         "allowed_physical_gpus": [5, 7],
     }
     validate(launch, "resnet_p1_launch", "abc")
+    transformer_launch = copy.deepcopy(launch)
+    transformer_launch["num_jobs"] = 27
+    validate(transformer_launch, "transformer_p1_launch", "abc")
+    r2_launch = copy.deepcopy(transformer_launch)
+    r2_launch["stage"] = "r2"
+    validate(r2_launch, "resnet_r2_launch", "abc")
+    t2_launch = copy.deepcopy(transformer_launch)
+    t2_launch["stage"] = "t2"
+    validate(t2_launch, "transformer_t2_launch", "abc")
     bad_launch = copy.deepcopy(launch)
     bad_launch["allowed_physical_gpus"] = [0, 1]
     must_fail(bad_launch, "resnet_p1_launch", "abc")