@@ -60,7 +60,7 @@ def __filter(cases_) -> typing.List[TestCase]:
6060 skipped_cases .append (case )
6161
6262 for case in cases [:]:
63- if "RDMA MPI" in case .trace and not ARG ( "rdma_mpi" ) :
63+ if "RDMA MPI" in case .trace :
6464 cases .remove (case )
6565 skipped_cases .append (case )
6666
@@ -189,10 +189,7 @@ def _handle_case(case: TestCase, devices: typing.Set[int]):
189189 cons .print (f" [bold magenta]{ case .get_uuid ()} [/bold magenta] SKIP { case .trace } " )
190190 return
191191
192- if "RDMA MPI" in case .trace :
193- cmd = case .run ([PRE_PROCESS , SIMULATION ], gpus = devices , rdma_mpi = True )
194- else :
195- cmd = case .run ([PRE_PROCESS , SIMULATION ], gpus = devices )
192+ cmd = case .run ([PRE_PROCESS , SIMULATION ], gpus = devices )
196193
197194 out_filepath = os .path .join (case .get_dirpath (), "out_pre_sim.txt" )
198195
@@ -236,9 +233,6 @@ def _handle_case(case: TestCase, devices: typing.Set[int]):
236233
237234 if ARG ("test_all" ):
238235 case .delete_output ()
239- # if ARG("rdma_mpi"):
240- # cmd = case.run([PRE_PROCESS, SIMULATION, POST_PROCESS], gpus=devices, rdma_mpi=True)
241- # else:
242236 cmd = case .run ([PRE_PROCESS , SIMULATION , POST_PROCESS ], gpus = devices )
243237 out_filepath = os .path .join (case .get_dirpath (), "out_post.txt" )
244238 common .file_write (out_filepath , cmd .stdout )
0 commit comments