Skip to content

GitLab

Switch branch/tag
Find file BlameHistoryPermalink
  • 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
test_dist.py 815 Bytes