Skip to content

Commit

Permalink
Refactor primitive test to support multiple GPUs in rings (#94)
Browse files Browse the repository at this point in the history
* Refactor primitive test to support multiple GPUs in rings

* Make GPUs sync before transfer optional

* Use same ring format as RCCL

* Extend to 8 GPUs and report errors if there is no P2P access

* Control GPUs sync before ops from command line with "-s" option

* Change buffer size through command line option "-n"

Rename iterations command line option to "-i"
  • Loading branch information
wenkaidu authored Jul 5, 2019
1 parent 949d680 commit 70804da
Showing 1 changed file with 265 additions and 202 deletions.
Loading

0 comments on commit 70804da

Please sign in to comment.