.travis.yml 3.67 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
18
        - IDX = cpu
        - 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
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
        - TOOLKIT="cudatoolkit=${CUDA_SHORT}"
    # - name: macosx/cpu
    #   os: osx
    #   osx_image: xcode11.3
    #   language: shell
    #   env:
    #     - IDX=cpu
    #     - TORCH=${TORCH_VERSION}
    # - 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
75
76
77
78
79
80
81
    # - 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
82
    #     - TORCH=${TORCH_VERSION}+100
rusty1s's avatar
rusty1s committed
83
84
85
    #     - PATH=/c/Python37:/c/Python37/Scripts:$PATH
    #   before_install:
    #     - choco install python --version 3.7.6
rusty1s's avatar
revert  
rusty1s committed
86
    # - name: windows/cu102
rusty1s's avatar
rusty1s committed
87
88
89
90
91
92
93
94
95
96
    #   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
97

rusty1s's avatar
fix  
rusty1s committed
98
install:
rusty1s's avatar
update  
rusty1s committed
99
  # - source script/install.sh
rusty1s's avatar
rusty1s committed
100
  - source script/conda.sh
rusty1s's avatar
fix  
rusty1s committed
101
  - conda create --yes -n test python=3.7.6
rusty1s's avatar
rusty1s committed
102
  - source activate test
rusty1s's avatar
typo  
rusty1s committed
103
104
105
  - conda install pytorch=${TORCH_VERSION} ${TOOLKIT} -c pytorch --yes
  - pip install flake8
  - pip install codecov
rusty1s's avatar
rusty1s committed
106
107
108
109
110
  - python setup.py install
  # - pip3 install --upgrade pip
  # - pip3 install wheel
  # - pip3 install numpy
  # - pip3 install torch==${TORCH} -f https://download.pytorch.org/whl/torch_stable.html
rusty1s's avatar
rusty1s committed
111

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