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

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

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

jobs:
rusty1s's avatar
rusty1s committed
41
42
  # For daily builds, we only test on CUDA 10.2
  # For deployment, we exclude *all* macOS CUDA and Windows CUDA 9.2 jobs.
rusty1s's avatar
rusty1s committed
43
  exclude:
rusty1s's avatar
rusty1s committed
44
    # Python 3.8
rusty1s's avatar
update  
rusty1s committed
45
    - if: tag is blank and os != osx
rusty1s's avatar
rusty1s committed
46
      env: PYTHON_VERSION=3.8 TORCH_VERSION=1.6.0 IDX=cpu
rusty1s's avatar
update  
rusty1s committed
47
48
49
50
51
52
    - if: tag is blank or os != linux
      env: PYTHON_VERSION=3.8 TORCH_VERSION=1.6.0 IDX=cu92
    - if: tag is blank or os = osx
      env: PYTHON_VERSION=3.8 TORCH_VERSION=1.6.0 IDX=cu101
    - if: os = osx
      env: PYTHON_VERSION=3.8 TORCH_VERSION=1.6.0 IDX=cu102
rusty1s's avatar
rusty1s committed
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
    - if: tag is blank and os != osx
      env: PYTHON_VERSION=3.8 TORCH_VERSION=1.5.0 IDX=cpu
    - if: tag is blank or os != linux
      env: PYTHON_VERSION=3.8 TORCH_VERSION=1.5.0 IDX=cu92
    - if: tag is blank or os = osx
      env: PYTHON_VERSION=3.8 TORCH_VERSION=1.5.0 IDX=cu101
    - if: os = osx
      env: PYTHON_VERSION=3.8 TORCH_VERSION=1.5.0 IDX=cu102
    # Python 3.7
    - if: tag is blank
      env: PYTHON_VERSION=3.7 TORCH_VERSION=1.6.0 IDX=cpu
    - if: tag is blank or os != linux
      env: PYTHON_VERSION=3.7 TORCH_VERSION=1.6.0 IDX=cu92
    - if: tag is blank or os = osx
      env: PYTHON_VERSION=3.7 TORCH_VERSION=1.6.0 IDX=cu101
    - if: tag is blank or os = osx
      env: PYTHON_VERSION=3.7 TORCH_VERSION=1.6.0 IDX=cu102
    - if: tag is blank
      env: PYTHON_VERSION=3.7 TORCH_VERSION=1.5.0 IDX=cpu
    - if: tag is blank or os != linux
      env: PYTHON_VERSION=3.7 TORCH_VERSION=1.5.0 IDX=cu92
    - if: tag is blank or os = osx
      env: PYTHON_VERSION=3.7 TORCH_VERSION=1.5.0 IDX=cu101
    - if: tag is blank or os = osx
      env: PYTHON_VERSION=3.7 TORCH_VERSION=1.5.0 IDX=cu102


rusty1s's avatar
update  
rusty1s committed
80

rusty1s's avatar
rusty1s committed
81
82
83
84
    # - if: ($TRAVIS_TAG = "" && ! $TRAVIS_COMMIT_MESSAGE =~ ci-deploy) || $TRAVIS_OS_NAME = "osx"
    #   env: PYTHON_VERSION=3.8 TORCH_VERSION=1.6.0 IDX=cu101
    # - if: $TRAVIS_OS_NAME = "osx"
    #   env: PYTHON_VERSION=3.8 TORCH_VERSION=1.6.0 IDX=cu102
rusty1s's avatar
rusty1s committed
85

rusty1s's avatar
fix  
rusty1s committed
86
install:
rusty1s's avatar
rusty1s committed
87
  - source script/gcc.sh
rusty1s's avatar
oder  
rusty1s committed
88
  - source script/cuda.sh
rusty1s's avatar
rusty1s committed
89
  - source script/conda.sh
rusty1s's avatar
rusty1s committed
90
91
  - conda create --yes -n test python="${PYTHON_VERSION}"
  - source activate test
rusty1s's avatar
rusty1s committed
92
  - conda install pytorch=${TORCH_VERSION} ${TOOLKIT} -c pytorch --yes
rusty1s's avatar
rusty1s committed
93
  - source script/torch.sh
rusty1s's avatar
release  
rusty1s committed
94
95
  - pip install flake8
  - pip install codecov
rusty1s's avatar
rusty1s committed
96
  - pip install .[test]
rusty1s's avatar
typo  
rusty1s committed
97
script:
rusty1s's avatar
rusty1s committed
98
  - flake8 .
rusty1s's avatar
typo  
rusty1s committed
99
100
  - python setup.py test
after_success:
rusty1s's avatar
rusty1s committed
101
102
  - python setup.py bdist_wheel --dist-dir=dist/torch-${TORCH_VERSION}
  - python script/rename_wheel.py ${IDX}
rusty1s's avatar
rusty1s committed
103
  - codecov
rusty1s's avatar
update  
rusty1s committed
104
105
106
107
108
109
110
111
112
113
114
115
116
117
# 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_scatter
#     condition: $TRAVIS_TAG != "" || $TRAVIS_COMMIT_MESSAGE =~ ci-deploy
rusty1s's avatar
rusty1s committed
118
119
notifications:
  email: false