new build

Author: rusty1s

.travis.yml

@@ -1,21 +1,21 @@
 language: shell
 
 os:
-  # - linux
+  - linux
   # - osx
   - windows
 
 env:
   global:
     - CUDA_HOME=/usr/local/cuda
   jobs:
-    - TORCH_VERSION=1.4.0 PYTHON_VERSION=3.8 IDX=cpu
+    # - TORCH_VERSION=1.4.0 PYTHON_VERSION=3.8 IDX=cpu
     # - TORCH_VERSION=1.4.0 PYTHON_VERSION=3.8 IDX=cu92
     # - TORCH_VERSION=1.4.0 PYTHON_VERSION=3.8 IDX=cu100
     # - TORCH_VERSION=1.4.0 PYTHON_VERSION=3.8 IDX=cu101
     # - TORCH_VERSION=1.4.0 PYTHON_VERSION=3.7 IDX=cpu
     - TORCH_VERSION=1.4.0 PYTHON_VERSION=3.7 IDX=cu92
-    - TORCH_VERSION=1.4.0 PYTHON_VERSION=3.7 IDX=cu100
+    # - TORCH_VERSION=1.4.0 PYTHON_VERSION=3.7 IDX=cu100
     - TORCH_VERSION=1.4.0 PYTHON_VERSION=3.7 IDX=cu101
     # - TORCH_VERSION=1.4.0 PYTHON_VERSION=3.6 IDX=cpu
     # - TORCH_VERSION=1.4.0 PYTHON_VERSION=3.6 IDX=cu92

script/cuda.sh

@@ -64,13 +64,13 @@ else
   export FORCE_CUDA=1
 fi
 
-if [ "${IDX}" == "cu100" ] || [ "${IDX}" == "cu101" ]; then
-  export NVCC_FLAGS="-gencode=arch=compute_35,code=sm_35 -gencode=arch=compute_50,code=sm_50 -gencode=arch=compute_60,code=sm_60 -gencode=arch=compute_70,code=sm_70 -gencode=arch=compute_75,code=sm_75 -gencode=arch=compute_50,code=compute_50"
-fi
+# if [ "${IDX}" == "cu100" ] || [ "${IDX}" == "cu101" ]; then
+#   export NVCC_FLAGS="-gencode=arch=compute_35,code=sm_35 -gencode=arch=compute_50,code=sm_50 -gencode=arch=compute_60,code=sm_60 -gencode=arch=compute_70,code=sm_70 -gencode=arch=compute_75,code=sm_75 -gencode=arch=compute_50,code=compute_50"
+# fi
 
-if [ "${IDX}" == "cu92" ]; then
-  export NVCC_FLAGS="-gencode=arch=compute_35,code=sm_35 -gencode=arch=compute_50,code=sm_50 -gencode=arch=compute_60,code=sm_60 -gencode=arch=compute_70,code=sm_70 -gencode=arch=compute_50,code=compute_50"
-fi
+# if [ "${IDX}" == "cu92" ]; then
+#   export NVCC_FLAGS="-gencode=arch=compute_35,code=sm_35 -gencode=arch=compute_50,code=sm_50 -gencode=arch=compute_60,code=sm_60 -gencode=arch=compute_70,code=sm_70 -gencode=arch=compute_50,code=compute_50"
+# fi
 
 if [ "${TRAVIS_OS_NAME}" = "linux" ] && [ "${IDX}" != "cpu" ]; then
   INSTALLER=cuda-repo-${UBUNTU_VERSION}_${CUDA}_amd64.deb

setup.py

@@ -26,7 +26,7 @@ def get_extensions():
         define_macros += [('WITH_CUDA', None)]
         nvcc_flags = os.getenv('NVCC_FLAGS', '')
         nvcc_flags = [] if nvcc_flags == '' else nvcc_flags.split(' ')
-        nvcc_flags += ['--expt-relaxed-constexpr']
+        nvcc_flags += ['-arch=sm_35', '--expt-relaxed-constexpr']
         extra_compile_args['nvcc'] = nvcc_flags
 
     extensions_dir = osp.join(osp.dirname(osp.abspath(__file__)), 'csrc')