.travis.yml 3.18 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
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
    # 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.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
update  
rusty1s committed
44
    - if: tag is blank and os != osx
rusty1s's avatar
rusty1s committed
45
      env: PYTHON_VERSION=3.8 TORCH_VERSION=1.6.0 IDX=cpu
rusty1s's avatar
update  
rusty1s committed
46
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
    # - 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
57

rusty1s's avatar
fix  
rusty1s committed
58
install:
rusty1s's avatar
rusty1s committed
59
  - source script/gcc.sh
rusty1s's avatar
oder  
rusty1s committed
60
  - source script/cuda.sh
rusty1s's avatar
rusty1s committed
61
  - source script/conda.sh
rusty1s's avatar
rusty1s committed
62
63
  - conda create --yes -n test python="${PYTHON_VERSION}"
  - source activate test
rusty1s's avatar
rusty1s committed
64
  - conda install pytorch=${TORCH_VERSION} ${TOOLKIT} -c pytorch --yes
rusty1s's avatar
rusty1s committed
65
  - source script/torch.sh
rusty1s's avatar
release  
rusty1s committed
66
67
  - pip install flake8
  - pip install codecov
rusty1s's avatar
rusty1s committed
68
  - pip install .[test]
rusty1s's avatar
typo  
rusty1s committed
69
script:
rusty1s's avatar
rusty1s committed
70
  - flake8 .
rusty1s's avatar
typo  
rusty1s committed
71
72
  - python setup.py test
after_success:
rusty1s's avatar
rusty1s committed
73
74
  - python setup.py bdist_wheel --dist-dir=dist/torch-${TORCH_VERSION}
  - python script/rename_wheel.py ${IDX}
rusty1s's avatar
rusty1s committed
75
  - codecov
rusty1s's avatar
update  
rusty1s committed
76
77
78
79
80
81
82
83
84
85
86
87
88
89
# 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
90
91
notifications:
  email: false