Pulls out save_validation_info boolean logic

rafvasq · rafvasq · commit 0e23e92bd161 · 2025-11-18T15:17:27.000-05:00
Signed-off-by: Rafael Vasquez &lt;rafvasq21@gmail.com&gt;
diff --git a/aiu_fms_testing_utils/scripts/drive_paged_programs.py b/aiu_fms_testing_utils/scripts/drive_paged_programs.py
@@ -812,6 +812,9 @@ def main():
     distributed_kwargs = get_distributed_kwargs(
         args.distributed, args.dist_timeout, args.save_validation_info_outputs
     )
+    save_validation_info_outputs = save_validation_info_outputs and (
+        dist.get_rank() == 0
+    )
 
     model = load_model(
         device_type="cpu",

Original file line number	Diff line number	Diff line change
`@@ -812,6 +812,9 @@ def main():`
`812`	`812`	`distributed_kwargs = get_distributed_kwargs(`
`813`	`813`	`args.distributed, args.dist_timeout, args.save_validation_info_outputs`
`814`	`814`	`)`
	`815`	`+ save_validation_info_outputs = save_validation_info_outputs and (`
	`816`	`+ dist.get_rank() == 0`
	`817`	`+ )`
`815`	`818`
`816`	`819`	`model = load_model(`
`817`	`820`	`device_type="cpu",`