test pytorch 1.6.0 build

065cf15b · rusty1s · f0e951cd · 065cf15b · 065cf15b · 065cf15b
Commit 065cf15b authored Jul 31, 2020 by rusty1s
Show whitespace changes
Inline Side-by-side

Showing with 105 additions and 39 deletions

.travis.yml .travis.yml +81 -38

README.md README.md +16 -0

script/torch.sh script/torch.sh +8 -1

No files found.
--- a/.travis.yml
+++ b/.travis.yml
@@ -8,47 +8,90 @@ os:
 env:
  global:
    - CUDA_HOME=/usr/local/cuda
-
  jobs:
-    - TORCH_VERSION=1.5.0 PYTHON_VERSION=3.8 IDX=cpu
-    - TORCH_VERSION=1.5.0 PYTHON_VERSION=3.8 IDX=cu92
-    - TORCH_VERSION=1.5.0 PYTHON_VERSION=3.8 IDX=cu101
-    - TORCH_VERSION=1.5.0 PYTHON_VERSION=3.8 IDX=cu102
-    - TORCH_VERSION=1.5.0 PYTHON_VERSION=3.7 IDX=cpu
-    - TORCH_VERSION=1.5.0 PYTHON_VERSION=3.7 IDX=cu92
-    - TORCH_VERSION=1.5.0 PYTHON_VERSION=3.7 IDX=cu101
-    - TORCH_VERSION=1.5.0 PYTHON_VERSION=3.7 IDX=cu102
-    - TORCH_VERSION=1.5.0 PYTHON_VERSION=3.6 IDX=cpu
-    - TORCH_VERSION=1.5.0 PYTHON_VERSION=3.6 IDX=cu92
-    - TORCH_VERSION=1.5.0 PYTHON_VERSION=3.6 IDX=cu101
-    - TORCH_VERSION=1.5.0 PYTHON_VERSION=3.6 IDX=cu102
+    # Python 3.6
+    - PYTHON_VERSION=3.6 TORCH_VERSION=1.6.0 IDX=cpu
+    - PYTHON_VERSION=3.6 TORCH_VERSION=1.6.0 IDX=cu92
+    - PYTHON_VERSION=3.6 TORCH_VERSION=1.6.0 IDX=cu101
+    - PYTHON_VERSION=3.6 TORCH_VERSION=1.6.0 IDX=cu102
+    - PYTHON_VERSION=3.6 TORCH_VERSION=1.5.0 IDX=cpu
+    - PYTHON_VERSION=3.6 TORCH_VERSION=1.5.0 IDX=cu92
+    - PYTHON_VERSION=3.6 TORCH_VERSION=1.5.0 IDX=cu101
+    - PYTHON_VERSION=3.6 TORCH_VERSION=1.5.0 IDX=cu102
+    # Python 3.7
+    - PYTHON_VERSION=3.7 TORCH_VERSION=1.6.0 IDX=cpu
+    - PYTHON_VERSION=3.7 TORCH_VERSION=1.6.0 IDX=cu92
+    - PYTHON_VERSION=3.7 TORCH_VERSION=1.6.0 IDX=cu101
+    - PYTHON_VERSION=3.7 TORCH_VERSION=1.6.0 IDX=cu102
+    - PYTHON_VERSION=3.7 TORCH_VERSION=1.5.0 IDX=cpu
+    - PYTHON_VERSION=3.7 TORCH_VERSION=1.5.0 IDX=cu92
+    - PYTHON_VERSION=3.7 TORCH_VERSION=1.5.0 IDX=cu101
+    - PYTHON_VERSION=3.7 TORCH_VERSION=1.5.0 IDX=cu102
+    # Python 3.8
+    - PYTHON_VERSION=3.8 TORCH_VERSION=1.6.0 IDX=cpu
+    - PYTHON_VERSION=3.8 TORCH_VERSION=1.6.0 IDX=cu92
+    - PYTHON_VERSION=3.8 TORCH_VERSION=1.6.0 IDX=cu101
+    - PYTHON_VERSION=3.8 TORCH_VERSION=1.6.0 IDX=cu102
+    - PYTHON_VERSION=3.8 TORCH_VERSION=1.5.0 IDX=cpu
+    - PYTHON_VERSION=3.8 TORCH_VERSION=1.5.0 IDX=cu92
+    - PYTHON_VERSION=3.8 TORCH_VERSION=1.5.0 IDX=cu101
+    - PYTHON_VERSION=3.8 TORCH_VERSION=1.5.0 IDX=cu102

 jobs:
-  exclude:  # Exclude *all* macOS CUDA jobs and Windows CUDA 9.2 jobs.
-    - os: osx
-      env: TORCH_VERSION=1.5.0 PYTHON_VERSION=3.8 IDX=cu92
-    - os: osx
-      env: TORCH_VERSION=1.5.0 PYTHON_VERSION=3.8 IDX=cu101
-    - os: osx
-      env: TORCH_VERSION=1.5.0 PYTHON_VERSION=3.8 IDX=cu102
-    - os: osx
-      env: TORCH_VERSION=1.5.0 PYTHON_VERSION=3.7 IDX=cu92
-    - os: osx
-      env: TORCH_VERSION=1.5.0 PYTHON_VERSION=3.7 IDX=cu101
-    - os: osx
-      env: TORCH_VERSION=1.5.0 PYTHON_VERSION=3.7 IDX=cu102
-    - os: osx
-      env: TORCH_VERSION=1.5.0 PYTHON_VERSION=3.6 IDX=cu92
-    - os: osx
-      env: TORCH_VERSION=1.5.0 PYTHON_VERSION=3.6 IDX=cu101
-    - os: osx
-      env: TORCH_VERSION=1.5.0 PYTHON_VERSION=3.6 IDX=cu102
-    - os: windows
-      env: TORCH_VERSION=1.5.0 PYTHON_VERSION=3.8 IDX=cu92
-    - os: windows
-      env: TORCH_VERSION=1.5.0 PYTHON_VERSION=3.7 IDX=cu92
-    - os: windows
-      env: TORCH_VERSION=1.5.0 PYTHON_VERSION=3.6 IDX=cu92
+  # For daily builds, we only test on *Python 3.6* and *CUDA 10.2*.
+  # For deployment, we exclude *all* macOS CUDA and Windows CUDA 9.2 jobs.
+  exclude:
+    # Python 3.6
+    - if: not (tag is present or commit_message =~ ci-deploy) and os != osx
+      env: PYTHON_VERSION=3.6 TORCH_VERSION=1.6.0 IDX=cpu
+    - if: not (tag is present or commit_message =~ ci-deploy) or os != linux
+      env: PYTHON_VERSION=3.6 TORCH_VERSION=1.6.0 IDX=cu92
+    - if: not (tag is present or commit_message =~ ci-deploy) or os = osx
+      env: PYTHON_VERSION=3.6 TORCH_VERSION=1.6.0 IDX=cu101
+    - if: os = osx
+      env: PYTHON_VERSION=3.6 TORCH_VERSION=1.6.0 IDX=cu102
+    - if: not (tag is present or commit_message =~ ci-deploy) and os != osx
+      env: PYTHON_VERSION=3.6 TORCH_VERSION=1.5.0 IDX=cpu
+    - if: not (tag is present or commit_message =~ ci-deploy) or os != linux
+      env: PYTHON_VERSION=3.6 TORCH_VERSION=1.5.0 IDX=cu92
+    - if: not (tag is present or commit_message =~ ci-deploy) or os = osx
+      env: PYTHON_VERSION=3.6 TORCH_VERSION=1.5.0 IDX=cu101
+    - if: os = osx
+      env: PYTHON_VERSION=3.6 TORCH_VERSION=1.5.0 IDX=cu102
+    # Python 3.7
+    - if: not (tag is present or commit_message =~ ci-deploy)
+      env: PYTHON_VERSION=3.7 TORCH_VERSION=1.6.0 IDX=cpu
+    - if: not (tag is present or commit_message =~ ci-deploy) or os != linux
+      env: PYTHON_VERSION=3.7 TORCH_VERSION=1.6.0 IDX=cu92
+    - if: not (tag is present or commit_message =~ ci-deploy) or os = osx
+      env: PYTHON_VERSION=3.7 TORCH_VERSION=1.6.0 IDX=cu101
+    - if: not (tag is present or commit_message =~ ci-deploy) or os = osx
+      env: PYTHON_VERSION=3.7 TORCH_VERSION=1.6.0 IDX=cu102
+    - if: not (tag is present or commit_message =~ ci-deploy)
+      env: PYTHON_VERSION=3.7 TORCH_VERSION=1.5.0 IDX=cpu
+    - if: not (tag is present or commit_message =~ ci-deploy) or os != linux
+      env: PYTHON_VERSION=3.7 TORCH_VERSION=1.5.0 IDX=cu92
+    - if: not (tag is present or commit_message =~ ci-deploy) or os = osx
+      env: PYTHON_VERSION=3.7 TORCH_VERSION=1.5.0 IDX=cu101
+    - if: not (tag is present or commit_message =~ ci-deploy) or os = osx
+      env: PYTHON_VERSION=3.7 TORCH_VERSION=1.5.0 IDX=cu102
+    # Python 3.8
+    - if: not (tag is present or commit_message =~ ci-deploy)
+      env: PYTHON_VERSION=3.8 TORCH_VERSION=1.6.0 IDX=cpu
+    - if: not (tag is present or commit_message =~ ci-deploy) or os != linux
+      env: PYTHON_VERSION=3.8 TORCH_VERSION=1.6.0 IDX=cu92
+    - if: not (tag is present or commit_message =~ ci-deploy) or os = osx
+      env: PYTHON_VERSION=3.8 TORCH_VERSION=1.6.0 IDX=cu101
+    - if: not (tag is present or commit_message =~ ci-deploy) or os = osx
+      env: PYTHON_VERSION=3.8 TORCH_VERSION=1.6.0 IDX=cu102
+    - if: not (tag is present or commit_message =~ ci-deploy)
+      env: PYTHON_VERSION=3.8 TORCH_VERSION=1.5.0 IDX=cpu
+    - if: not (tag is present or commit_message =~ ci-deploy) or os != linux
+      env: PYTHON_VERSION=3.8 TORCH_VERSION=1.5.0 IDX=cu92
+    - if: not (tag is present or commit_message =~ ci-deploy) or os = osx
+      env: PYTHON_VERSION=3.8 TORCH_VERSION=1.5.0 IDX=cu101
+    - if: not (tag is present or commit_message =~ ci-deploy) or os = osx
+      env: PYTHON_VERSION=3.8 TORCH_VERSION=1.5.0 IDX=cu102

 install:
  - source script/gcc.sh

--- a/README.md
+++ b/README.md
@@ -25,6 +25,22 @@ The operator works on all floating point data types and is implemented both for

 We provide pip wheels for all major OS/PyTorch/CUDA combinations, see [here](https://pytorch-geometric.com/whl).

+#### PyTorch 1.6.0
+
+To install the binaries for PyTorch 1.6.0, simply run
+
+```
+pip install torch-spline-conv==latest+${CUDA} -f https://pytorch-geometric.com/whl/torch-1.6.0.html
+```
+
+where `${CUDA}` should be replaced by either `cpu`, `cu92`, `cu101` or `cu102` depending on your PyTorch installation.
+
+|             | `cpu` | `cu92` | `cu101` | `cu102` |
+|-------------|-------|--------|---------|---------|
+| **Linux**   | ✅    | ✅     | ✅      | ✅      |
+| **Windows** | ✅    | ❌     | ✅      | ✅      |
+| **macOS**   | ✅    |        |         |         |
+
 #### PyTorch 1.5.0

 To install the binaries for PyTorch 1.5.0, simply run

--- a/script/torch.sh
+++ b/script/torch.sh
 #!/bin/bash

 # Fix "member may not be initialized" error on Windows: https://github.com/pytorch/pytorch/issues/27958
-if [ "${TRAVIS_OS_NAME}" = "windows" ] && [ "${IDX}" != "cpu" ]; then
+if [ "${TRAVIS_OS_NAME}" = "windows" ] && [ "${TORCH_VERSION}" = "1.5.0" ]; then
+  echo "Fix nvcc for PyTorch 1.5.0"
  sed -i.bak -e 's/constexpr/const/g' /c/tools/miniconda3/envs/test/lib/site-packages/torch/include/torch/csrc/jit/api/module.h
  sed -i.bak -e 's/constexpr/const/g' /c/tools/miniconda3/envs/test/lib/site-packages/torch/include/torch/csrc/jit/runtime/argument_spec.h
  sed -i.bak -e 's/return \*(this->value)/return \*((type\*)this->value)/g' /c/tools/miniconda3/envs/test/lib/site-packages/torch/include/pybind11/cast.h
 fi

+# https://github.com/pytorch/pytorch/commit/d2e16dd888a9b5fd55bd475d4fcffb70f388d4f0
+if [ "${TRAVIS_OS_NAME}" = "windows" ] && [ "${TORCH_VERSION}" = "1.6.0" ]; then
+  echo "Fix nvcc for PyTorch 1.6.0"
+  sed -i.bak -e 's/CONSTEXPR_EXCEPT_WIN_CUDA/const/g' /c/tools/miniconda3/envs/test/lib/site-packages/torch/include/torch/csrc/jit/api/module.h
+  sed -i.bak -e 's/return \*(this->value)/return \*((type\*)this->value)/g' /c/tools/miniconda3/envs/test/lib/site-packages/torch/include/pybind11/cast.h
+fi