diff options
Diffstat (limited to 'src/transport/shm.cc')
-rw-r--r-- | src/transport/shm.cc | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/src/transport/shm.cc b/src/transport/shm.cc index 60f16c8..0b1d8ee 100644 --- a/src/transport/shm.cc +++ b/src/transport/shm.cc @@ -104,7 +104,7 @@ ncclResult_t shmRecvSetup(struct ncclTopoSystem* topo, struct ncclTopoGraph* gra } /* Connect to this peer */ -ncclResult_t shmSendConnect(struct ncclConnect* connectInfo, struct ncclConnector* send) { +ncclResult_t shmSendConnect(struct ncclConnect* connectInfo, int nranks, int rank, struct ncclConnector* send) { // Setup device pointers struct shmConnectInfo* info = (struct shmConnectInfo*)connectInfo; struct shmSendResources* resources = (struct shmSendResources*)send->transportResources; @@ -129,7 +129,7 @@ ncclResult_t shmSendConnect(struct ncclConnect* connectInfo, struct ncclConnecto return ncclSuccess; } -ncclResult_t shmRecvConnect(struct ncclConnect* connectInfo, struct ncclConnector* recv) { +ncclResult_t shmRecvConnect(struct ncclConnect* connectInfo, int nranks, int rank, struct ncclConnector* recv) { // Setup device pointers struct shmRecvResources* resources = (struct shmRecvResources*)recv->transportResources; struct shmConnectInfo* info = (struct shmConnectInfo*)connectInfo; |