Change __CUDACC_VER_*__ preprocessor directives to CUDA_VERSION because clang doesn't define the former.

이 커밋은 다음에 포함됨:
Christian Sigg
2018-12-13 15:59:22 +01:00
커밋한 사람 Sylvain Jeaugey
부모 469b69a5d0
커밋 d08e9b5279
4개의 변경된 파일6개의 추가작업 그리고 6개의 파일을 삭제
+1 -1
파일 보기
@@ -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 __CUDACC_VER_MAJOR__ >= 9
#if CUDA_VERSION >= 9000
if (cgMode & 0x01) {
CUDACHECK(cudaLaunchCooperativeKernelMultiDevice(paramsList, numDevices,
// These flags are to reduce the latency of using this API