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

rusty1s's avatar
matrix  
rusty1s committed
3
env:
rusty1s's avatar
rusty1s committed
4
  global:
rusty1s's avatar
rusty1s committed
5
    - TORCH_VERSION=1.4.0
rusty1s's avatar
update  
rusty1s committed
6
    - CUDA_HOME=/usr/local/cuda
rusty1s's avatar
rusty1s committed
7
8
9
  # jobs:
  #   - FORCE_CUDA=0 TORCH=${TORCH_VERSION}+cpu
  #   - FORCE_CUDA=1 CUDA_SHORT=9.2 CUDA=9.2.148-1 UBUNTU_VERSION=ubuntu1604 CUBLAS=cuda-cublas-dev-9-2 TORCH=${TORCH_VERSION}+cu92
rusty1s's avatar
rusty1s committed
10
    # - FORCE_CUDA=1 CUDA_SHORT=10.0 CUDA=10.0.130-1 UBUNTU_VERSION=ubuntu1804 CUBLAS=cuda-cublas-dev-10-0 TORCH=${TORCH_VERSION}+cu100
rusty1s's avatar
rusty1s committed
11
12

jobs:
rusty1s's avatar
rusty1s committed
13
  include:
rusty1s's avatar
rusty1s committed
14
15
16
    - name: linux/cpu
      os: linux
      env:
rusty1s's avatar
rusty1s committed
17
        - IDX=cpu
rusty1s's avatar
rusty1s committed
18
        - TOOLKIT=cpuonly
rusty1s's avatar
rusty1s committed
19
20
21
22
23
24
25
26
    - name: linux/cu92
      os: linux
      env:
        - IDX=cu92
        - CUDA_SHORT=9.2
        - CUDA=9.2.148-1
        - UBUNTU_VERSION=ubuntu1604
        - CUBLAS=cuda-cublas-dev-9-2
rusty1s's avatar
rusty1s committed
27
        - TOOLKIT="cudatoolkit=${CUDA_SHORT}"
rusty1s's avatar
rusty1s committed
28
29
30
    - name: linux/cu100
      os: linux
      env:
rusty1s's avatar
update  
rusty1s committed
31
        - IDX=cu100
rusty1s's avatar
rusty1s committed
32
33
34
35
        - CUDA_SHORT=10.0
        - CUDA=10.0.130-1
        - UBUNTU_VERSION=ubuntu1804
        - CUBLAS=cuda-cublas-dev-10-0
rusty1s's avatar
rusty1s committed
36
        - TOOLKIT="cudatoolkit=${CUDA_SHORT}"
rusty1s's avatar
rusty1s committed
37
38
39
40
41
42
43
44
    - name: linux/cu101
      os: linux
      env:
        - IDX=cu101
        - CUDA_SHORT=10.1
        - CUDA=10.1.105-1
        - UBUNTU_VERSION=ubuntu1804
        - CUBLAS=libcublas-dev
rusty1s's avatar
rusty1s committed
45
        - TOOLKIT="cudatoolkit=${CUDA_SHORT}"
rusty1s's avatar
rusty1s committed
46
47
48
49
50
    - name: macosx/cpu
      os: osx
      env:
        - IDX=cpu
        - TOOLKIT=""
rusty1s's avatar
rusty1s committed
51
52
53
54
55
    - name: windows/cpu
      os: windows
      env:
        - IDX=cpu
        - TOOLKIT=cpuonly
rusty1s's avatar
rusty1s committed
56
57
58
59
60
61
62
63
64
65
66
67
    # - name: windows/cu92
    #   os: windows
    #   language: shell
    #   env:
    #     - IDX=cu92
    #     - CUDA_SHORT=9.2
    #     - CUDA=9.2.148
    #     - TORCH=${TORCH_VERSION}+${IDX}
    #     - PATH=/c/Python37:/c/Python37/Scripts:$PATH
    #   before_install:
    #     - choco install python --version 3.7.6
    #     - python -m pip install --upgrade --user pip
rusty1s's avatar
rusty1s committed
68
69
70
71
72
73
74
    # - name: windows/cu100
    #   os: windows
    #   language: shell
    #   env:
    #     - FORCE_CUDA=1
    #     - CUDA_SHORT=10.1
    #     - CUDA=10.0.130_411.31
rusty1s's avatar
fix  
rusty1s committed
75
    #     - TORCH=${TORCH_VERSION}+100
rusty1s's avatar
rusty1s committed
76
77
78
    #     - PATH=/c/Python37:/c/Python37/Scripts:$PATH
    #   before_install:
    #     - choco install python --version 3.7.6
rusty1s's avatar
revert  
rusty1s committed
79
    # - name: windows/cu102
rusty1s's avatar
rusty1s committed
80
81
82
83
84
85
86
87
88
89
    #   os: windows
    #   language: shell
    #   env:
    #     - FORCE_CUDA=1
    #     - CUDA_SHORT=10.1
    #     - CUDA=10.1.105_418.96
    #     - TORCH=${TORCH_VERSION}
    #     - PATH=/c/Python37:/c/Python37/Scripts:$PATH
    #   before_install:
    #     - choco install python --version 3.7.6
rusty1s's avatar
matrix  
rusty1s committed
90

rusty1s's avatar
fix  
rusty1s committed
91
install:
rusty1s's avatar
rusty1s committed
92
  - source script/install.sh
rusty1s's avatar
rusty1s committed
93
  - source script/conda.sh
rusty1s's avatar
rusty1s committed
94
  - conda update --yes conda
rusty1s's avatar
fix  
rusty1s committed
95
  - conda create --yes -n test python=3.7.6
rusty1s's avatar
rusty1s committed
96
  - source activate test
rusty1s's avatar
typo  
rusty1s committed
97
98
99
  - conda install pytorch=${TORCH_VERSION} ${TOOLKIT} -c pytorch --yes
  - pip install flake8
  - pip install codecov
rusty1s's avatar
rusty1s committed
100
  - python setup.py install
rusty1s's avatar
rusty1s committed
101

rusty1s's avatar
typo  
rusty1s committed
102
103
104
105
script:
  - flake8 .
  - python setup.py test
after_success:
rusty1s's avatar
rusty1s committed
106
  - python setup.py bdist_wheel --dist-dir=dist/torch-${TORCH_VERSION}/${IDX}
rusty1s's avatar
typo  
rusty1s committed
107
108
109
  - codecov
deploy:
  provider: s3
rusty1s's avatar
rusty1s committed
110
  region: eu-central-1
rusty1s's avatar
typo  
rusty1s committed
111
112
113
114
  edge: true
  access_key_id: AKIAJB7S6NJ5OM5MAAGA
  secret_access_key: ${S3_SECRET_ACCESS_KEY}
  bucket: pytorch-scatter
rusty1s's avatar
rusty1s committed
115
116
  local_dir: dist/torch-${TORCH_VERSION}/${IDX}
  upload_dir: whl/torch-${TORCH_VERSION}/${IDX}
rusty1s's avatar
typo  
rusty1s committed
117
118
119
120
121
  acl: public_read
  on:
    repo: rusty1s/pytorch_scatter
    # tags: true
    branch: travis
rusty1s's avatar
rusty1s committed
122
123
notifications:
  email: false