.travis.yml 2.95 KB
Newer Older
rusty1s's avatar
rusty1s committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
language: shell

os:
  - 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=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=cu101
    - TORCH_VERSION=1.4.0 PYTHON_VERSION=3.6 IDX=cpu
    - TORCH_VERSION=1.4.0 PYTHON_VERSION=3.6 IDX=cu92
    - TORCH_VERSION=1.4.0 PYTHON_VERSION=3.6 IDX=cu100
    - TORCH_VERSION=1.4.0 PYTHON_VERSION=3.6 IDX=cu101

rusty1s's avatar
rusty1s committed
25
jobs:
rusty1s's avatar
rusty1s committed
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
  exclude:  # Exclude *all* macOS CUDA jobs and Windows CUDA 9.2/10.0 jobs.
    - os: osx
      env: TORCH_VERSION=1.4.0 PYTHON_VERSION=3.8 IDX=cu92
    - os: osx
      env: TORCH_VERSION=1.4.0 PYTHON_VERSION=3.8 IDX=cu100
    - os: osx
      env: TORCH_VERSION=1.4.0 PYTHON_VERSION=3.8 IDX=cu101
    - os: osx
      env: TORCH_VERSION=1.4.0 PYTHON_VERSION=3.7 IDX=cu92
    - os: osx
      env: TORCH_VERSION=1.4.0 PYTHON_VERSION=3.7 IDX=cu100
    - os: osx
      env: TORCH_VERSION=1.4.0 PYTHON_VERSION=3.7 IDX=cu101
    - os: osx
      env: TORCH_VERSION=1.4.0 PYTHON_VERSION=3.6 IDX=cu92
    - os: osx
      env: TORCH_VERSION=1.4.0 PYTHON_VERSION=3.6 IDX=cu100
    - os: osx
      env: TORCH_VERSION=1.4.0 PYTHON_VERSION=3.6 IDX=cu101
    - os: windows
      env: TORCH_VERSION=1.4.0 PYTHON_VERSION=3.8 IDX=cu92
    - os: windows
      env: TORCH_VERSION=1.4.0 PYTHON_VERSION=3.8 IDX=cu100
    - os: windows
      env: TORCH_VERSION=1.4.0 PYTHON_VERSION=3.7 IDX=cu92
    - os: windows
      env: TORCH_VERSION=1.4.0 PYTHON_VERSION=3.7 IDX=cu100
    - os: windows
      env: TORCH_VERSION=1.4.0 PYTHON_VERSION=3.6 IDX=cu92
    - os: windows
      env: TORCH_VERSION=1.4.0 PYTHON_VERSION=3.6 IDX=cu100
rusty1s's avatar
rusty1s committed
57
58
    - os: windows
      env: TORCH_VERSION=1.4.0 PYTHON_VERSION=3.8 IDX=cu101
rusty1s's avatar
rusty1s committed
59

rusty1s's avatar
rusty1s committed
60
install:
rusty1s's avatar
rusty1s committed
61
62
63
64
65
66
  - source script/cuda.sh
  - source script/conda.sh
  - 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
67
  - pip install torch-scatter==latest+${IDX} -f https://s3.eu-central-1.amazonaws.com/pytorch-geometric.com/whl/torch-1.4.0.html
rusty1s's avatar
rusty1s committed
68
69
  - pip install flake8 codecov
  - python setup.py install
rusty1s's avatar
rusty1s committed
70
script:
rusty1s's avatar
rusty1s committed
71
  - flake8 .
rusty1s's avatar
rusty1s committed
72
73
  - python setup.py test
after_success:
rusty1s's avatar
rusty1s committed
74
75
  - python setup.py bdist_wheel --dist-dir=dist/torch-${TORCH_VERSION}
  - python script/rename_wheel.py ${IDX}
rusty1s's avatar
rusty1s committed
76
  - codecov
rusty1s's avatar
rusty1s committed
77
78
79
80
81
82
deploy:
  provider: s3
  region: eu-central-1
  edge: true
  access_key_id: AKIAJB7S6NJ5OM5MAAGA
  secret_access_key: ${S3_SECRET_ACCESS_KEY}
rusty1s's avatar
rusty1s committed
83
  bucket: pytorch-geometric.com
rusty1s's avatar
rusty1s committed
84
85
86
87
88
  local_dir: dist/torch-${TORCH_VERSION}
  upload_dir: whl/torch-${TORCH_VERSION}
  acl: public_read
  on:
    repo: rusty1s/pytorch_sparse
rusty1s's avatar
rusty1s committed
89
    tags: true
rusty1s's avatar
rusty1s committed
90
91
notifications:
  email: false