[REVIEW] Make NCCL root initialization configurable. (#120)

* Initial fixes * Add worker root node ability, and unit tests * Add logging and improve verbose usage * Fix typos, add better docs info * More logging updates * Update changelog * Fix test_comms.py style issues * Style updates * Style updates * Style updates * Feedback updates * Feedback updates - consolidate code, merge worker_state and scheduler_state functions * Consolidate test code paths * Style updates. * Add default state object to get_raft_comm_state to improve cuml compatibility * Style fix Co-authored-by: Devin Robison <[email protected]>
rapidsai · Jan 20, 2021 · 87efed0 · 87efed0
1 parent b5454cf
commit 87efed0
Show file tree

Hide file tree

Showing 4 changed files with 473 additions and 143 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -23,6 +23,7 @@
 - PR #103: Epsilon parameter for Cholesky rank one update
 - PR #100: Add divyegala as codeowner
 - PR #111: Cleanup gpuCI scripts
+- PR #120: Update NCCL init process to support root node placement.
 
 ## Bug Fixes
 - PR #106: Specify dependency branches to avoid pip resolver failure