Merge pull request #692 from yaeliyac/UCC_add_perftest_generator_a2av

ucc_perftest_add_gen_and_a2av

Author: amaslenn

src/cloudai/workloads/ucc_test/slurm_command_gen_strategy.py

@@ -39,6 +39,8 @@ def generate_test_command(self) -> List[str]:
         srun_command_parts.append(f"-c {tdef_cmd_args.collective}")
         srun_command_parts.append(f"-b {tdef_cmd_args.b}")
         srun_command_parts.append(f"-e {tdef_cmd_args.e}")
+        if tdef_cmd_args.gen is not None:
+            srun_command_parts.append(f"--gen {tdef_cmd_args.gen}")
         srun_command_parts.append("-m cuda")
         srun_command_parts.append("-F")
 

src/cloudai/workloads/ucc_test/ucc.py

@@ -68,6 +68,7 @@ class UCCCmdArgs(CmdArgs):
     ] = "alltoall"
     b: Union[int, list[int]] = 1
     e: Union[str, list[str]] = "8M"
+    gen: Union[str, list[str], None] = None
 
 
 class UCCTestDefinition(TestDefinition):