Enable GPU direct RDMA read from GPU

[ROCm/rccl commit: 5f97122442]
2021-02-03 02:48:30 +00:00
parent 6bf9b0d36a
commit 57abf599b2
4 changed files with 27 additions and 4 deletions
@@ -115,6 +115,7 @@ struct ncclConnInfo {
  // allows software to explicitly initiate a flush read to HDP memory. See more
  // descriptions in primitives.h.
  uint32_t* next_hdp_reg;  // Next GPU in ring (for p2p transport use only)
+  uint32_t* curr_hdp_reg;  // Current GPU's HDP register
 };

 struct ncclConnector {