Replace CUDA_VERSION by CUDART_VERSION
Tá an tiomantas seo le fáil i:
@@ -58,7 +58,7 @@ static void* const ncclKerns[ncclCollCount*ncclNumOps*ncclNumTypes*2] = {
|
||||
};
|
||||
|
||||
ncclResult_t ncclLaunchCooperativeKernelMultiDevice(struct cudaLaunchParams *paramsList, int* cudaDevs, int numDevices, int cgMode) {
|
||||
#if CUDA_VERSION >= 9000
|
||||
#if CUDART_VERSION >= 9000
|
||||
if (cgMode & 0x01) {
|
||||
CUDACHECK(cudaLaunchCooperativeKernelMultiDevice(paramsList, numDevices,
|
||||
// These flags are to reduce the latency of using this API
|
||||
|
||||
Tagairt in Eagrán Nua
Cuir bac ar úsáideoir