pass devlist as const int* rather than int* in ncclCommInitAll

2 files changed, 2 insertions, 2 deletions

diff --git a/src/core.cu b/src/core.cu
index f3d1462..ac0d65f 100644
--- a/src/core.cu
+++ b/src/core.cu
@@ -750,7 +750,7 @@ ncclResult_t ncclCommInitRank(ncclComm_t* newcomm, int ndev, ncclUniqueId commId
 }
 
 extern "C" DSOGLOBAL
-ncclResult_t ncclCommInitAll(ncclComm_t* comms, int ndev, int* devlist) {
+ncclResult_t ncclCommInitAll(ncclComm_t* comms, int ndev, const int* devlist) {
   initDebug();
 
   showVersion();
diff --git a/src/nccl.h b/src/nccl.h
index a0a71fc..47bbdd8 100644
--- a/src/nccl.h
+++ b/src/nccl.h
@@ -85,7 +85,7 @@ ncclResult_t ncclCommInitRank(ncclComm_t* comm, int ndev, ncclUniqueId commId, i
  * comm should be pre-allocated with size at least ndev*sizeof(ncclComm_t).
  * If devlist is NULL, the first ndev CUDA devices are used.
  * Order of devlist defines user-order of processors within the communicator. */
-ncclResult_t ncclCommInitAll(ncclComm_t* comm, int ndev, int* devlist);
+ncclResult_t ncclCommInitAll(ncclComm_t* comm, int ndev, const int* devlist);
 
 /* Frees resources associated with communicator object. */
 void ncclCommDestroy(ncclComm_t comm);