- 05 Feb, 2020 1 commit
-
-
Shaden Smith authored
* Enables NCCL backend in @distributed_test * Adds pytest-forked to avoid CUDA re-initialization issue. * paste typo * transcription typo
-
- 04 Feb, 2020 2 commits
-
-
Jeff Rasley authored
* add allreduce test * comment out set rank to cuda for now * switched back to gloo
-
Shaden Smith authored
* Adds distributed_test decorator and some unit tests. * Setting NCCL backend. * Parametrizes test. * rank -> local_rank * Temporarily disable CUDA initialization.
-