Replace CUDA_VERSION by CUDART_VERSION

[ROCm/rccl commit: c244b51ae7]
Tento commit je obsažen v:
Sylvain Jeaugey
2018-12-13 15:22:17 -08:00
rodič 6a93dca51b
revize 4e32c7a19a
4 změnil soubory, kde provedl 6 přidání a 6 odebrání
+1 -1
Zobrazit soubor
@@ -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