Skip to content

GitLab

Switch branch/tag
  1. 04 Feb, 2020 2 commits
    • Jeff Rasley's avatar
      add allreduce test (#7) · 52c5a936
      Jeff Rasley authored 
      * add allreduce test

* comment out set rank to cuda for now

* switched back to gloo
      52c5a936
    • Shaden Smith's avatar
      Distributed testing (#6) · b61a2217
      Shaden Smith authored 
      * Adds distributed_test decorator and some unit tests.

* Setting NCCL backend.

* Parametrizes test.

* rank -> local_rank

* Temporarily disable CUDA initialization.
      b61a2217