language: shell

os:
  - linux
  - osx
  - windows

env:
  global:
    - CUDA_HOME=/usr/local/cuda
  jobs:
    - TORCH_VERSION=1.5.0 PYTHON_VERSION=3.8 IDX=cpu
    - TORCH_VERSION=1.5.0 PYTHON_VERSION=3.8 IDX=cu92
    - TORCH_VERSION=1.5.0 PYTHON_VERSION=3.8 IDX=cu101
    - TORCH_VERSION=1.5.0 PYTHON_VERSION=3.8 IDX=cu102
    - TORCH_VERSION=1.5.0 PYTHON_VERSION=3.7 IDX=cpu
    - TORCH_VERSION=1.5.0 PYTHON_VERSION=3.7 IDX=cu92
    - TORCH_VERSION=1.5.0 PYTHON_VERSION=3.7 IDX=cu101
    - TORCH_VERSION=1.5.0 PYTHON_VERSION=3.7 IDX=cu102
    - TORCH_VERSION=1.5.0 PYTHON_VERSION=3.6 IDX=cpu
    - TORCH_VERSION=1.5.0 PYTHON_VERSION=3.6 IDX=cu92
    - TORCH_VERSION=1.5.0 PYTHON_VERSION=3.6 IDX=cu101
    - TORCH_VERSION=1.5.0 PYTHON_VERSION=3.6 IDX=cu102

jobs:
  exclude:  # Exclude *all* macOS CUDA jobs and Windows CUDA 9.2 jobs.
    - os: osx
      env: TORCH_VERSION=1.5.0 PYTHON_VERSION=3.8 IDX=cu92
    - os: osx
      env: TORCH_VERSION=1.5.0 PYTHON_VERSION=3.8 IDX=cu101
    - os: osx
      env: TORCH_VERSION=1.5.0 PYTHON_VERSION=3.8 IDX=cu102
    - os: osx
      env: TORCH_VERSION=1.5.0 PYTHON_VERSION=3.7 IDX=cu92
    - os: osx
      env: TORCH_VERSION=1.5.0 PYTHON_VERSION=3.7 IDX=cu101
    - os: osx
      env: TORCH_VERSION=1.5.0 PYTHON_VERSION=3.7 IDX=cu102
    - os: osx
      env: TORCH_VERSION=1.5.0 PYTHON_VERSION=3.6 IDX=cu92
    - os: osx
      env: TORCH_VERSION=1.5.0 PYTHON_VERSION=3.6 IDX=cu101
    - os: osx
      env: TORCH_VERSION=1.5.0 PYTHON_VERSION=3.6 IDX=cu102
    - os: windows
      env: TORCH_VERSION=1.5.0 PYTHON_VERSION=3.8 IDX=cu92
    - os: windows
      env: TORCH_VERSION=1.5.0 PYTHON_VERSION=3.7 IDX=cu92
    - os: windows
      env: TORCH_VERSION=1.5.0 PYTHON_VERSION=3.6 IDX=cu92

install:
  - source script/gcc.sh
  - source script/cuda.sh
  - source script/conda.sh
  - source script/metis.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 https://s3.eu-central-1.amazonaws.com/pytorch-geometric.com/whl/torch-1.5.0.html
  - pip install flake8 codecov
  - python setup.py install
script:
  - flake8 .
  - python setup.py test
after_success:
  - python setup.py bdist_wheel --dist-dir=dist/torch-${TORCH_VERSION}
  - python script/rename_wheel.py ${IDX}
  - codecov
deploy:
  provider: s3
  region: eu-central-1
  edge: true
  access_key_id: ${S3_ACCESS_KEY}
  secret_access_key: ${S3_SECRET_ACCESS_KEY}
  bucket: pytorch-geometric.com
  local_dir: dist/torch-${TORCH_VERSION}
  upload_dir: whl/torch-${TORCH_VERSION}
  acl: public_read
  on:
    all_branches: true
    repo: rusty1s/pytorch_scatter
    condition: $TRAVIS_TAG != "" || $TRAVIS_COMMIT_MESSAGE =~ ci-deploy
notifications:
  email: false