Comments out CUDA_ARCH setting for compilation - Test

apache · Mar 29, 2019 · 5480ed5 · 5480ed5
1 parent c01d5f7
commit 5480ed5
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/ci/docker/runtime_functions.sh b/ci/docker/runtime_functions.sh
@@ -347,7 +347,7 @@ build_centos7_gpu() {
         USE_CUDA_PATH=/usr/local/cuda             \
         USE_CUDNN=1                               \
         USE_DIST_KVSTORE=1                        \
-        CUDA_ARCH="$CI_CUDA_COMPUTE_CAPABILITIES" \
+        # CUDA_ARCH="$CI_CUDA_COMPUTE_CAPABILITIES" \
         -j$(nproc)
 }
 
@@ -643,7 +643,7 @@ build_ubuntu_gpu_mkldnn() {
         USE_CUDA=1                                \
         USE_CUDA_PATH=/usr/local/cuda             \
         USE_CUDNN=1                               \
-        CUDA_ARCH="$CI_CUDA_COMPUTE_CAPABILITIES" \
+        # CUDA_ARCH="$CI_CUDA_COMPUTE_CAPABILITIES" \
         USE_SIGNAL_HANDLER=1                      \
         -j$(nproc)
 }
@@ -660,7 +660,7 @@ build_ubuntu_gpu_mkldnn_nocudnn() {
         USE_CUDA=1                                \
         USE_CUDA_PATH=/usr/local/cuda             \
         USE_CUDNN=0                               \
-        CUDA_ARCH="$CI_CUDA_COMPUTE_CAPABILITIES" \
+        # CUDA_ARCH="$CI_CUDA_COMPUTE_CAPABILITIES" \
         USE_SIGNAL_HANDLER=1                      \
         -j$(nproc)
 }
@@ -679,7 +679,7 @@ build_ubuntu_gpu_cuda100_cudnn7() {
         USE_CUDNN=1                               \
         USE_CPP_PACKAGE=1                         \
         USE_DIST_KVSTORE=1                        \
-        CUDA_ARCH="$CI_CUDA_COMPUTE_CAPABILITIES" \
+        # CUDA_ARCH="$CI_CUDA_COMPUTE_CAPABILITIES" \
         USE_SIGNAL_HANDLER=1                      \
         -j$(nproc)
 }