Merge branch 'ko3n1g/ci/fail-fast-unit-tests' into 'main'

ci: Fail-fast on unit tests See merge request ADLR/megatron-lm!2520
NVIDIA · Jan 8, 2025 · 1ce944c · 1ce944c
2 parents 05780f3 + 9220838
commit 1ce944c
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