name: Compatibility Test

on: workflow_dispatch

jobs:
  build:
    name: Test for PyTorch compatibility
    if: github.base_ref == 'main' && github.repository == 'hpcaitech/ColossalAI'
    runs-on: [self-hosted, gpu]
    strategy:
      fail-fast: false
      matrix:
        # pytorch-cuda:1.9.0-11.1.1 + Python 3.8
        # pytorch-cuda:1.8.1-11.1.1 + Python 3.8
        # pytorch-cuda:1.7.1-11.0.3 + Python 3.8
        # pytorch-cuda:1.6.0-10.2 + Python 3.6
        container: ["frankleeeee/pytorch-cuda:1.9.0-11.1.1",
                    "frankleeeee/pytorch-cuda:1.8.1-11.1.1",
                    "frankleeeee/pytorch-cuda:1.7.1-11.0.3",
                    "frankleeeee/pytorch-cuda:1.6.0-10.2"]
    container:
      image: ${{ matrix.container }}
      options: --gpus all --rm --ipc=host -v /data/scratch/cifar-10:/data/scratch/cifar-10
    timeout-minutes: 120
    steps:
      - name: Install dependencies
        run: |
          pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple
          pip install -U pip setuptools wheel --user
          pip install pytest tensorboard
      - uses: actions/checkout@v2
      - name: Install Colossal-AI      
        run: |
          pip install -r requirements/requirements.txt
          pip install -v --no-cache-dir .
      - name: Unit Testing
        run: |
          PYTHONPATH=$PWD pytest tests
        env:
          DATA: /data/scratch/cifar-10