ADLR/megatron-lm!2520 - ci: Fail-fast on unit tests

NVIDIA · Jan 8, 2025 · 9220838 · 9220838
1 parent 4dc8977
commit 9220838
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/tests/test_utils/python_scripts/launch_jet_workload.py b/tests/test_utils/python_scripts/launch_jet_workload.py
@@ -271,6 +271,9 @@ def main(
         success = pipeline.get_status() == PipelineStatus.SUCCESS
         logger.info("Pipeline terminated with status %s", pipeline.get_status().name)
 
+        if test_type == "unit_test":
+            sys.exit(int(not success))  # invert for exit 0
+
         if test_type != "release":
             if success:
                 sys.exit(int(not success))  # invert for exit 0