Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
.travis.yml 2.95 KB
Newer Older
rusty1s's avatar
build wheels  
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
maco os travis  
rusty1s committed
25 
jobs:
rusty1s's avatar
build wheels  
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
57
58 
  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
initial commit  
rusty1s committed
59 
install:
rusty1s's avatar
build wheels  
rusty1s committed
60
61
62
63
64
65
66
67
68 
  - 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
  - pip install torch-scatter==latest+${IDX} -f http://pytorch-scatter.s3-website.eu-central-1.amazonaws.com/whl/torch-${TORCH_VERSION}.html --trusted-host pytorch-scatter.s3-website.eu-central-1.amazonaws.com
  - pip install flake8 codecov
  - python setup.py install
rusty1s's avatar
initial commit  
rusty1s committed
69 
script:
rusty1s's avatar
removed py2.7 test  
rusty1s committed
70 
  - flake8 .
rusty1s's avatar
initial commit  
rusty1s committed
71
72 
  - python setup.py test
after_success:
rusty1s's avatar
build wheels  
rusty1s committed
73
74 
  - python setup.py bdist_wheel --dist-dir=dist/torch-${TORCH_VERSION}
  - python script/rename_wheel.py ${IDX}
rusty1s's avatar
initial commit  
rusty1s committed
75 
  - codecov
rusty1s's avatar
build wheels  
rusty1s committed
76
77
78
79
80
81
82
83
84
85
86
87 
deploy:
  provider: s3
  region: eu-central-1
  edge: true
  access_key_id: AKIAJB7S6NJ5OM5MAAGA
  secret_access_key: ${S3_SECRET_ACCESS_KEY}
  bucket: pytorch-sparse
  local_dir: dist/torch-${TORCH_VERSION}
  upload_dir: whl/torch-${TORCH_VERSION}
  acl: public_read
  on:
    repo: rusty1s/pytorch_sparse
rusty1s's avatar
only deploy on tags  
rusty1s committed
88 
    tags: true
rusty1s's avatar
initial commit  
rusty1s committed
89
90 
notifications:
  email: false