Fix the issue in case CUPY_NO_DUDA is defined
@@ -40,6 +40,16 @@ ncclResult_t ncclBcast(...);
ncclResult_t ncclReduceScatter(...);
ncclResult_t ncclAllGather(...);
+typedef enum {
+ ncclChar = 0,
+ ncclInt = 1,
+ ncclHalf = 2,
+ ncclFloat = 3,
+ ncclDouble = 4,
+ ncclInt64 = 5,
+ ncclUint64 = 6,
+ nccl_NUM_TYPES = 7 } ncclDataType_t;
+
}
#endif // #ifndef CUPY_NO_CUDA
