From 9220838787d09aa69e4b9877cc17542d16ea0b8d Mon Sep 17 00:00:00 2001 From: Oliver Koenig Date: Tue, 7 Jan 2025 18:12:23 -0800 Subject: [PATCH] ADLR/megatron-lm!2520 - ci: Fail-fast on unit tests --- tests/test_utils/python_scripts/launch_jet_workload.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/tests/test_utils/python_scripts/launch_jet_workload.py b/tests/test_utils/python_scripts/launch_jet_workload.py index 43185c7cf1..c663571966 100644 --- 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