ci-deploy PyTorch 1.8.0 wheels

Author: rusty1s

.travis.yml

@@ -14,31 +14,31 @@ env:
     - PYTHON_VERSION=3.6 TORCH_VERSION=1.8.0 IDX=cu101
     - PYTHON_VERSION=3.6 TORCH_VERSION=1.8.0 IDX=cu102
     - PYTHON_VERSION=3.6 TORCH_VERSION=1.8.0 IDX=cu111
-    - PYTHON_VERSION=3.6 TORCH_VERSION=1.7.0 IDX=cpu
-    - PYTHON_VERSION=3.6 TORCH_VERSION=1.7.0 IDX=cu92
-    - PYTHON_VERSION=3.6 TORCH_VERSION=1.7.0 IDX=cu101
-    - PYTHON_VERSION=3.6 TORCH_VERSION=1.7.0 IDX=cu102
-    - PYTHON_VERSION=3.6 TORCH_VERSION=1.7.0 IDX=cu110
+    # - PYTHON_VERSION=3.6 TORCH_VERSION=1.7.0 IDX=cpu
+    # - PYTHON_VERSION=3.6 TORCH_VERSION=1.7.0 IDX=cu92
+    # - PYTHON_VERSION=3.6 TORCH_VERSION=1.7.0 IDX=cu101
+    # - PYTHON_VERSION=3.6 TORCH_VERSION=1.7.0 IDX=cu102
+    # - PYTHON_VERSION=3.6 TORCH_VERSION=1.7.0 IDX=cu110
     # Python 3.7
     - PYTHON_VERSION=3.7 TORCH_VERSION=1.8.0 IDX=cpu
     - PYTHON_VERSION=3.7 TORCH_VERSION=1.8.0 IDX=cu101
     - PYTHON_VERSION=3.7 TORCH_VERSION=1.8.0 IDX=cu102
     - PYTHON_VERSION=3.7 TORCH_VERSION=1.8.0 IDX=cu111
-    - PYTHON_VERSION=3.7 TORCH_VERSION=1.7.0 IDX=cpu
-    - PYTHON_VERSION=3.7 TORCH_VERSION=1.7.0 IDX=cu92
-    - PYTHON_VERSION=3.7 TORCH_VERSION=1.7.0 IDX=cu101
-    - PYTHON_VERSION=3.7 TORCH_VERSION=1.7.0 IDX=cu102
-    - PYTHON_VERSION=3.7 TORCH_VERSION=1.7.0 IDX=cu110
+    # - PYTHON_VERSION=3.7 TORCH_VERSION=1.7.0 IDX=cpu
+    # - PYTHON_VERSION=3.7 TORCH_VERSION=1.7.0 IDX=cu92
+    # - PYTHON_VERSION=3.7 TORCH_VERSION=1.7.0 IDX=cu101
+    # - PYTHON_VERSION=3.7 TORCH_VERSION=1.7.0 IDX=cu102
+    # - PYTHON_VERSION=3.7 TORCH_VERSION=1.7.0 IDX=cu110
     # Python 3.8
     - PYTHON_VERSION=3.8 TORCH_VERSION=1.8.0 IDX=cpu
     - PYTHON_VERSION=3.8 TORCH_VERSION=1.8.0 IDX=cu101
     - PYTHON_VERSION=3.8 TORCH_VERSION=1.8.0 IDX=cu102
     - PYTHON_VERSION=3.8 TORCH_VERSION=1.8.0 IDX=cu111
-    - PYTHON_VERSION=3.8 TORCH_VERSION=1.7.0 IDX=cpu
-    - PYTHON_VERSION=3.8 TORCH_VERSION=1.7.0 IDX=cu92
-    - PYTHON_VERSION=3.8 TORCH_VERSION=1.7.0 IDX=cu101
-    - PYTHON_VERSION=3.8 TORCH_VERSION=1.7.0 IDX=cu102
-    - PYTHON_VERSION=3.8 TORCH_VERSION=1.7.0 IDX=cu110
+    # - PYTHON_VERSION=3.8 TORCH_VERSION=1.7.0 IDX=cpu
+    # - PYTHON_VERSION=3.8 TORCH_VERSION=1.7.0 IDX=cu92
+    # - PYTHON_VERSION=3.8 TORCH_VERSION=1.7.0 IDX=cu101
+    # - PYTHON_VERSION=3.8 TORCH_VERSION=1.7.0 IDX=cu102
+    # - PYTHON_VERSION=3.8 TORCH_VERSION=1.7.0 IDX=cu110
 
 jobs:
   # For daily builds, we only test on Python 3.6 with 11.0/11.1.

setup.py

@@ -10,9 +10,6 @@
 from torch.utils.cpp_extension import BuildExtension
 from torch.utils.cpp_extension import CppExtension, CUDAExtension, CUDA_HOME
 
-major, minor = int(str(torch.__version__)[0]), int(str(torch.__version__)[2])
-use_ninja = major >= 2 or (major == 1 and minor >= 8)
-
 WITH_CUDA = torch.cuda.is_available() and CUDA_HOME is not None
 suffices = ['cpu', 'cuda'] if WITH_CUDA else ['cpu']
 if os.getenv('FORCE_CUDA', '0') == '1':
@@ -99,8 +96,7 @@ def get_extensions():
     ext_modules=get_extensions() if not BUILD_DOCS else [],
     cmdclass={
         'build_ext':
-        BuildExtension.with_options(no_python_abi_suffix=True,
-                                    use_ninja=use_ninja)
+        BuildExtension.with_options(no_python_abi_suffix=True, use_ninja=False)
     },
     packages=find_packages(),
 )