Replace CUDA_VERSION by CUDART_VERSION

Tá an tiomantas seo le fáil i:
Sylvain Jeaugey
2018-12-13 15:22:17 -08:00
tuismitheoir 3e6afef473
tiomantas c244b51ae7
D'athraigh 4 comhad le 6 breiseanna agus 6 scriosta
+1 -1
Féach ar an gComhad
@@ -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