accelerate
3e0ceac7 - Attempt to fix import error when PyTorch is build without `torch.distributed` module (#1108)

Commit
2 years ago
Attempt to fix import error when PyTorch is build without `torch.distributed` module (#1108) * Attempt to fix importing invalid `torch.distributed.ReduceOp` when torch is built without distributed support. * Style. * Move `torch.distributed` logic detection to `imports.py` according to @muellerzr comments * Style. * Update wording * Remove raising exceptions in the case of a non-distributed setup, simply dont import the ReduceOp in this case.
Author
Parents
Loading