Skip to content

Commit 75be19c

Browse files
committed
Print ompi_info
1 parent 4569c89 commit 75be19c

1 file changed

Lines changed: 5 additions & 1 deletion

File tree

deepspeed/launcher/multinode_runner.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -85,7 +85,11 @@ def __init__(self, args, world_info_base64, resource_pool):
8585

8686
def backend_exists(self):
8787
#TODO: if IB is available we should suggestion mvapich
88-
return shutil.which('ompi_info')
88+
exists = shutil.which('ompi_info')
89+
if not exists:
90+
warnings.warn("ompi_info does not exist, openmpi is not installed properly")
91+
92+
return exists
8993

9094
def get_cmd(self, environment, active_resources):
9195
#TODO: Allow for include/exclude at node-level but not gpu-level

0 commit comments

Comments
 (0)