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

os:
rusty1s's avatar
rusty1s committed
4
5
  # - linux
  # - osx
rusty1s's avatar
rusty1s committed
6
7
8
9
10
11
12
  - windows

env:
  global:
    - CUDA_HOME=/usr/local/cuda
  jobs:
    - TORCH_VERSION=1.4.0 PYTHON_VERSION=3.8 IDX=cpu
rusty1s's avatar
rusty1s committed
13
14
15
16
17
18
19
20
21
22
23
    # - 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
24

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
  - conda install -c anaconda metis --yes
rusty1s's avatar
rusty1s committed
68
  - 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
69
70
  - pip install flake8 codecov
  - python setup.py install
rusty1s's avatar
rusty1s committed
71
script:
rusty1s's avatar
rusty1s committed
72
  - flake8 .
rusty1s's avatar
rusty1s committed
73
74
  - python setup.py test
after_success:
rusty1s's avatar
rusty1s committed
75
76
  - python setup.py bdist_wheel --dist-dir=dist/torch-${TORCH_VERSION}
  - python script/rename_wheel.py ${IDX}
rusty1s's avatar
rusty1s committed
77
  - codecov
rusty1s's avatar
rusty1s committed
78
79
80
81
deploy:
  provider: s3
  region: eu-central-1
  edge: true
rusty1s's avatar
rusty1s committed
82
  access_key_id: ${S3_ACCESS_KEY}
rusty1s's avatar
rusty1s committed
83
  secret_access_key: ${S3_SECRET_ACCESS_KEY}
rusty1s's avatar
rusty1s committed
84
  bucket: pytorch-geometric.com
rusty1s's avatar
rusty1s committed
85
86
87
88
89
  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
90
    tags: true
rusty1s's avatar
rusty1s committed
91
92
notifications:
  email: false