Use hipGraph instead of cudaGraph (#487)

[ROCm/rccl commit: 3669e12432]
This commit is contained in:
Wenkai Du
2022-01-10 08:26:01 -08:00
committed by GitHub
parent 3f16835565
commit 123de1621f
4 changed files with 6 additions and 8 deletions
+1 -3
View File
@@ -15,8 +15,6 @@
// [/RCCL]
#if defined(__HIP_PLATFORM_HCC__) || defined(__HCC__) || defined(__HIPCC__)
typedef void *cudaGraph_t;
typedef void *cudaGraphNode_t;
#define HIPRT_CB
#else
#include "collectives.h"
@@ -216,7 +214,7 @@ struct ncclComm {
struct ncclQueueInfo* enqueueInfo;
int nQueueInfoCreated;
int nQueueInfoDestroyed;
cudaGraphNode_t lastSetupNode;
hipGraphNode_t lastSetupNode;
unsigned long long lastCudaGraphId;
int driverVersion;
pfn_cuMemGetAddressRange_t pfnCuMemGetAddressRange;