diff --git a/projects/rccl/src/enqueue.cc b/projects/rccl/src/enqueue.cc index 15915fe19e..95035a0483 100644 --- a/projects/rccl/src/enqueue.cc +++ b/projects/rccl/src/enqueue.cc @@ -1101,7 +1101,7 @@ void* graphHelperFunc(void *args) { } } -ncclResult_t ncclGetCudaGraph(ncclComm_t comm, cudaGraph_t* graph) { +ncclResult_t ncclGetCudaGraph(ncclComm_t comm, hipGraph_t* graph) { comm->usingCudaGraph = 0; #if CUDART_VERSION >= 11030 hipStreamCaptureStatus captureStatus; @@ -1285,7 +1285,7 @@ ncclResult_t ncclEnqueueCheck(struct ncclInfo* info) { info->datatype, info->op, info->root, info->comm, info->comm->nRanks, info->stream); // Check whether we are in cuda graph mode - cudaGraph_t graph; + hipGraph_t graph; ncclComm_t comm = info->comm; NCCLCHECKGOTO(ncclGetCudaGraph(comm, &graph), ret, end); diff --git a/projects/rccl/src/group.cc b/projects/rccl/src/group.cc index 8a8d212a61..dee0ccc288 100644 --- a/projects/rccl/src/group.cc +++ b/projects/rccl/src/group.cc @@ -174,7 +174,7 @@ ncclResult_t ncclGroupEnd() { for (int i=0; i void HIPRT_CB ncclEnqueueHostSetup(void* arg); -ncclResult_t ncclGetCudaGraph(ncclComm_t comm, cudaGraph_t* graph); -ncclResult_t ncclCudaGraphHostSetup(ncclComm_t comm, cudaGraph_t graph); +ncclResult_t ncclGetCudaGraph(ncclComm_t comm, hipGraph_t* graph); +ncclResult_t ncclCudaGraphHostSetup(ncclComm_t comm, hipGraph_t graph); struct ncclBuffRegInfo { void* sendbuffsBase[NCCL_MAX_INTRA_RANKS];