Method run_2cpu

tests/parallel_launch.py:168–184

(self, target_file_name)

class TestMultipleWithGloo(unittest.TestCase):
    def run_2cpu(self, target_file_name):

        cluster, pod = get_cluster_from_args([0, 1])  # tmp use. for getting trainer_nranks()

        procs = start_local_trainers_cpu(
            cluster.trainers_endpoints(),
            training_script=target_file_name,
            training_script_args=[],
        )

        while True:
            alive = watch_local_trainers(procs, cluster.trainers_nranks())

            if not alive:
                print("Local procs complete, POD info:{}".format(pod))
                break
            time.sleep(3)