diff --git a/projects/rccl-tests/src/common.cu b/projects/rccl-tests/src/common.cu index 13d430af42..96c191d8b9 100644 --- a/projects/rccl-tests/src/common.cu +++ b/projects/rccl-tests/src/common.cu @@ -1401,15 +1401,9 @@ testResult_t run() { // Free off CUDA allocated memory for (int i=0; i= NCCL_VERSION(2,19,0) - if (sendbuffs[i]) NCCLCHECK(ncclMemFree((char*)sendbuffs[i])); - if (recvbuffs[i]) NCCLCHECK(ncclMemFree((char*)recvbuffs[i])); - if (datacheck) NCCLCHECK(ncclMemFree(expected[i])); -#else if (sendbuffs[i]) CUDACHECK(cudaFree((char*)sendbuffs[i])); if (recvbuffs[i]) CUDACHECK(cudaFree((char*)recvbuffs[i])); if (datacheck) CUDACHECK(cudaFree(expected[i])); -#endif } CUDACHECK(cudaFreeHost(delta)); #if NCCL_VERSION_CODE >= NCCL_VERSION(2,19,0)