Skip to content

Commit

Permalink
exclude mpi run from auto-detection of horovod (#8610)
Browse files Browse the repository at this point in the history
  • Loading branch information
awaelchli committed Aug 3, 2021
1 parent 1e08d31 commit e82c66b
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -857,7 +857,7 @@ def check_horovod(self):
@staticmethod
def has_horovodrun() -> bool:
"""Returns True if running with `horovodrun` using Gloo or OpenMPI."""
return "OMPI_COMM_WORLD_RANK" in os.environ or "HOROVOD_RANK" in os.environ
return _HOROVOD_AVAILABLE and ("OMPI_COMM_WORLD_RANK" in os.environ or "HOROVOD_RANK" in os.environ)

def update_device_type_if_ipu_plugin(self) -> None:
# This allows the poptorch.Options that are passed into the IPUPlugin to be the source of truth,
Expand Down

0 comments on commit e82c66b

Please sign in to comment.