name: Building Wheels

on: [workflow_dispatch]

jobs:

  wheel:
    runs-on: ${{ matrix.os }}

    strategy:
      fail-fast: false
      matrix:
        # os: [ubuntu-18.04, macos-10.15, windows-latest]
        os: [windows-latest]
        # python-version: [3.6, 3.7, 3.8, 3.9]
        python-version: [3.6]
        # torch-version: [1.8.0, 1.9.0]
        torch-version: [1.8.0]
        # cuda-version: ['cpu', 'cu101', 'cu102', 'cu111']
        cuda-version: ['cu101']
        exclude:
          - torch-version: 1.9.0
            cuda-version: 'cu101'
          - os: macos-10.15
            cuda-version: 'cu101'
          - os: macos-10.15
            cuda-version: 'cu102'
          - os: macos-10.15
            cuda-version: 'cu111'

    steps:
      - uses: actions/checkout@v2
      - name: Set up Python ${{ matrix.python-version }}
        uses: actions/setup-python@v2
        with:
          python-version: ${{ matrix.python-version }}

      - name: Update pip
        run: |
          pip install --upgrade pip

      - name: Install CUDA ${{ matrix.cuda-version }}
        if: ${{ matrix.cuda-version != 'cpu' }}
        run: |
          bash .github/workflows/cuda/${{ matrix.cuda-version }}-${{ runner.os }}.sh

      - name: Install PyTorch ${{ matrix.torch-version }}+${{ matrix.cuda-version }}
        if : ${{ runner.os != 'macOS' }}
        run: |
          pip install --no-cache-dir torch==${{ matrix.torch-version}}+${{ matrix.cuda-version }} -f https://download.pytorch.org/whl/torch_stable.html

      - name: Install default PyTorch ${{ matrix.torch-version }}
        if : ${{ runner.os == 'macOS' || (matrix.torch-version == '1.8.0' && matrix.cuda-version == 'cu102') }}
        run: |
          pip install --no-cache-dir torch==${{ matrix.torch-version}} -f https://download.pytorch.org/whl/torch_stable.html

      - name: Install main package for CPU
        if: ${{ matrix.cuda-version == 'cpu' }}
        run: |
          FORCE_ONLY_CPU=1 pip install -e .
        shell:
          bash

      - name: Install main package with CUDA
        if: ${{ matrix.cuda-version != 'cpu' }}
        run: |
          source .github/workflows/cuda/${{ matrix.cuda-version }}-${{ runner.os }}-env.sh
          FORCE_CUDA=1 pip install -e .
        shell:
          bash

      - name: Build wheel
        run: |
          pip install wheel
          python setup.py bdist_wheel --dist-dir=dist

      # - name: Upload wheel
      #   uses: shallwefootball/s3-upload-action@master
      #   with:
      #     aws_key_id: ${{ secrets.AWS_KEY_ID }}
      #     aws_secret_access_key: ${{ secrets.AWS_SECRET_ACCESS_KEY}}
      #     aws_bucket: 'pytorch-geometric.com'
      #     source_dir: 'dist'
      #     destination_dir: 'whl/torch-${{ matrix.torch-version }}+${{ matrix.cuda-version }}'