.travis.yml 3.45 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
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
    # - name: windows/cpu
    #   os: windows
    #   language: shell
    #   env:
    #     - IDX=cpu
    #     - 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
    # - 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
73
74
75
76
77
78
79
    # - 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
80
    #     - TORCH=${TORCH_VERSION}+100
rusty1s's avatar
rusty1s committed
81
82
83
    #     - PATH=/c/Python37:/c/Python37/Scripts:$PATH
    #   before_install:
    #     - choco install python --version 3.7.6
rusty1s's avatar
revert  
rusty1s committed
84
    # - name: windows/cu102
rusty1s's avatar
rusty1s committed
85
86
87
88
89
90
91
92
93
94
    #   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
95

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

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