From 1c7ce68631b3eeb4edaaa3980e0f7e1fd50b2a7a Mon Sep 17 00:00:00 2001 From: Zach Kimberg Date: Thu, 22 Aug 2019 11:29:47 -0700 Subject: [PATCH] Fix Nightly Maven GPU --- make/maven/maven_darwin_mkl.mk | 2 +- make/maven/maven_linux_cu90mkl.mk | 3 +-- make/maven/maven_linux_cu92mkl.mk | 3 +-- make/maven/maven_linux_mkl.mk | 2 +- 4 files changed, 4 insertions(+), 6 deletions(-) diff --git a/make/maven/maven_darwin_mkl.mk b/make/maven/maven_darwin_mkl.mk index 9bf3fc46ce0b..a7f2bdb027d4 100644 --- a/make/maven/maven_darwin_mkl.mk +++ b/make/maven/maven_darwin_mkl.mk @@ -77,7 +77,7 @@ USE_CUDNN = 0 # CUDA_ARCH := # whether use cuda runtime compiling for writing kernels in native language (i.e. Python) -ENABLE_CUDA_RTC = 0 +USE_NVRTC = 0 # use openmp for parallelization USE_OPENMP = 0 diff --git a/make/maven/maven_linux_cu90mkl.mk b/make/maven/maven_linux_cu90mkl.mk index 00cfd5dfa31d..e9ba46509973 100644 --- a/make/maven/maven_linux_cu90mkl.mk +++ b/make/maven/maven_linux_cu90mkl.mk @@ -79,9 +79,8 @@ USE_NCCL = 1 # CUDA_ARCH := # whether use cuda runtime compiling for writing kernels in native language (i.e. Python) -ENABLE_CUDA_RTC = 1 - USE_NVTX=1 +USE_NVRTC = 1 # use openmp for parallelization USE_OPENMP = 1 diff --git a/make/maven/maven_linux_cu92mkl.mk b/make/maven/maven_linux_cu92mkl.mk index 6ac920a12ff0..caa1c59c01d5 100644 --- a/make/maven/maven_linux_cu92mkl.mk +++ b/make/maven/maven_linux_cu92mkl.mk @@ -79,9 +79,8 @@ USE_NCCL = 1 # CUDA_ARCH := # whether use cuda runtime compiling for writing kernels in native language (i.e. Python) -ENABLE_CUDA_RTC = 1 - USE_NVTX=1 +USE_NVRTC = 1 # use openmp for parallelization USE_OPENMP = 1 diff --git a/make/maven/maven_linux_mkl.mk b/make/maven/maven_linux_mkl.mk index 10aee5f35a46..3c8534a7e2aa 100644 --- a/make/maven/maven_linux_mkl.mk +++ b/make/maven/maven_linux_mkl.mk @@ -76,7 +76,7 @@ USE_CUDNN = 0 # CUDA_ARCH := # whether use cuda runtime compiling for writing kernels in native language (i.e. Python) -ENABLE_CUDA_RTC = 0 +USE_NVRTC = 0 # use openmp for parallelization USE_OPENMP = 1