Waiting for MPI host 'test-mpi-trainjob-pqnw2-launcher-0-0.test-mpi-trainjob-pqnw2' to resolve
Warning: Permanently added '[test-mpi-trainjob-pqnw2-node-0-0.test-mpi-trainjob-pqnw2]:2222' (ECDSA) to the list of known hosts.
MPI COLLECTIVES PASSED device=cpu rank=0 world_size=2
MPI COLLECTIVES PASSED device=cpu rank=1 world_size=2