.travis.yml 6.21 KB
Newer Older
rusty1s's avatar
rusty1s committed
1
2
3
language: shell

os:
rusty1s's avatar
cmake  
rusty1s committed
4
  - linux
rusty1s's avatar
rusty1s committed
5
  - osx
rusty1s's avatar
rusty1s committed
6
7
  - windows

rusty1s's avatar
rusty1s committed
8
9
10
env:
  global:
    - CUDA_HOME=/usr/local/cuda
rusty1s's avatar
rusty1s committed
11
  jobs:
rusty1s's avatar
rusty1s committed
12
    # Python 3.6
rusty1s's avatar
rusty1s committed
13
14
15
16
17
    - 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
rusty1s's avatar
rusty1s committed
18
19
20
21
22
    - 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 3.7
rusty1s's avatar
rusty1s committed
23
24
25
26
27
    - 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
rusty1s's avatar
rusty1s committed
28
29
30
31
32
    - 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 3.8
rusty1s's avatar
rusty1s committed
33
34
35
36
37
    - 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
rusty1s's avatar
rusty1s committed
38
39
40
41
    - 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
rusty1s's avatar
rusty1s committed
42

rusty1s's avatar
rusty1s committed
43
jobs:
rusty1s's avatar
rusty1s committed
44
  # For daily builds, we only test on Python 3.6 with CUDA 10.2/11.0.
rusty1s's avatar
rusty1s committed
45
46
47
  # For deployment, we exclude *all* macOS CUDA and Windows CUDA 9.2 jobs.
  exclude:
    # Python 3.6
rusty1s's avatar
rusty1s committed
48
49
50
51
52
53
54
55
56
57
    - if: not (tag is present or commit_message =~ ci-deploy) and os != osx
      env: PYTHON_VERSION=3.6 TORCH_VERSION=1.7.0 IDX=cpu
    - if: not (tag is present or commit_message =~ ci-deploy) or os != linux
      env: PYTHON_VERSION=3.6 TORCH_VERSION=1.7.0 IDX=cu92
    - if: not (tag is present or commit_message =~ ci-deploy) or os = osx
      env: PYTHON_VERSION=3.6 TORCH_VERSION=1.7.0 IDX=cu101
    - if: os = osx
      env: PYTHON_VERSION=3.6 TORCH_VERSION=1.7.0 IDX=cu102
    - if: os = osx
      env: PYTHON_VERSION=3.6 TORCH_VERSION=1.7.0 IDX=cu110
rusty1s's avatar
rusty1s committed
58
59
60
61
62
63
64
65
    - 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
rusty1s's avatar
rusty1s committed
66
67
68
    # Python 3.7
    - if: not (tag is present or commit_message =~ ci-deploy)
      env: PYTHON_VERSION=3.7 TORCH_VERSION=1.7.0 IDX=cpu
rusty1s's avatar
rusty1s committed
69
    - if: not (tag is present or commit_message =~ ci-deploy) or os != linux
rusty1s's avatar
rusty1s committed
70
      env: PYTHON_VERSION=3.7 TORCH_VERSION=1.7.0 IDX=cu92
rusty1s's avatar
rusty1s committed
71
    - if: not (tag is present or commit_message =~ ci-deploy) or os = osx
rusty1s's avatar
rusty1s committed
72
73
74
75
76
      env: PYTHON_VERSION=3.7 TORCH_VERSION=1.7.0 IDX=cu101
    - if: not (tag is present or commit_message =~ ci-deploy) or os = osx
      env: PYTHON_VERSION=3.7 TORCH_VERSION=1.7.0 IDX=cu102
    - if: not (tag is present or commit_message =~ ci-deploy) or os = osx
      env: PYTHON_VERSION=3.7 TORCH_VERSION=1.7.0 IDX=cu110
rusty1s's avatar
rusty1s committed
77
78
79
80
81
82
83
84
    - 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
rusty1s's avatar
rusty1s committed
85
    # Python 3.8
rusty1s's avatar
rusty1s committed
86
    - if: not (tag is present or commit_message =~ ci-deploy)
rusty1s's avatar
rusty1s committed
87
      env: PYTHON_VERSION=3.8 TORCH_VERSION=1.7.0 IDX=cpu
rusty1s's avatar
rusty1s committed
88
    - if: not (tag is present or commit_message =~ ci-deploy) or os != linux
rusty1s's avatar
rusty1s committed
89
      env: PYTHON_VERSION=3.8 TORCH_VERSION=1.7.0 IDX=cu92
rusty1s's avatar
rusty1s committed
90
    - if: not (tag is present or commit_message =~ ci-deploy) or os = osx
rusty1s's avatar
rusty1s committed
91
      env: PYTHON_VERSION=3.8 TORCH_VERSION=1.7.0 IDX=cu101
rusty1s's avatar
rusty1s committed
92
    - if: not (tag is present or commit_message =~ ci-deploy) or os = osx
rusty1s's avatar
rusty1s committed
93
94
95
      env: PYTHON_VERSION=3.8 TORCH_VERSION=1.7.0 IDX=cu102
    - if: not (tag is present or commit_message =~ ci-deploy) or os = osx
      env: PYTHON_VERSION=3.8 TORCH_VERSION=1.7.0 IDX=cu110
rusty1s's avatar
rusty1s committed
96
97
98
99
100
101
102
103
    - 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
rusty1s's avatar
rusty1s committed
104

rusty1s's avatar
rusty1s committed
105
install:
rusty1s's avatar
rusty1s committed
106
  - source script/gcc.sh
rusty1s's avatar
rusty1s committed
107
108
  - source script/cuda.sh
  - source script/conda.sh
rusty1s's avatar
cmake  
rusty1s committed
109
  - source script/metis.sh
rusty1s's avatar
rusty1s committed
110
111
112
113
  - conda create --yes -n test python="${PYTHON_VERSION}"
  - source activate test
  - conda install pytorch=${TORCH_VERSION} ${TOOLKIT} -c pytorch --yes
  - source script/torch.sh
rusty1s's avatar
rusty1s committed
114
  - pip install torch-scatter -f https://pytorch-geometric.com/whl/torch-${TORCH_VERSION}+${IDX}.html
rusty1s's avatar
rusty1s committed
115
  - pip install flake8 codecov
rusty1s's avatar
rusty1s committed
116
  - pip install scipy==1.4.1
rusty1s's avatar
rusty1s committed
117
  - source script/install.sh
rusty1s's avatar
rusty1s committed
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
script:
  - flake8 .
  - python setup.py test
after_success:
  - python setup.py bdist_wheel --dist-dir=dist/torch-${TORCH_VERSION}
  - python script/rename_wheel.py ${IDX}
  - codecov
deploy:
  provider: s3
  region: eu-central-1
  edge: true
  access_key_id: ${S3_ACCESS_KEY}
  secret_access_key: ${S3_SECRET_ACCESS_KEY}
  bucket: pytorch-geometric.com
  local_dir: dist/torch-${TORCH_VERSION}
  upload_dir: whl/torch-${TORCH_VERSION}
  acl: public_read
  on:
    all_branches: true
    repo: rusty1s/pytorch_sparse
    condition: $TRAVIS_TAG != "" || $TRAVIS_COMMIT_MESSAGE =~ ci-deploy
notifications:
  email: false