Github Action Strategy Refactor (#97)

* Increase nightly regression test coverage

Have a config that that tries out different runners and pytorch versions

* Update regression_test.yml

---------

Co-authored-by: cpuhrsch <cpuhrsch@googlemail.com>

Author: msaroufim

.github/workflows/regression_test.yml

@@ -9,114 +9,38 @@ on:
       - main
 
 jobs:
-  test-cuda-2-2-2:
-    runs-on: 4-core-ubuntu-gpu-t4
+  test:
+    strategy:
+      matrix:
+        include:
+          - name: CUDA 2.2.2
+            runs-on: 4-core-ubuntu-gpu-t4
+            torch-spec: 'torch==2.2.2'
+          - name: CUDA 2.3 RC
+            runs-on: 4-core-ubuntu-gpu-t4
+            torch-spec: '--pre torch --index-url https://download.pytorch.org/whl/nightly/cu121'
+          - name: CUDA Nightly
+            runs-on: 4-core-ubuntu-gpu-t4
+            torch-spec: '--pre torch --index-url https://download.pytorch.org/whl/nightly/cu121'
+          - name: CPU
+            runs-on: 32-core-ubuntu
+            torch-spec: 'torch --index-url https://download.pytorch.org/whl/cpu'
+          - name: Nightly CPU
+            runs-on: 32-core-ubuntu
+            torch-spec: '--pre torch --index-url https://download.pytorch.org/whl/nightly/cpu'
+    runs-on: ${{ matrix.runs-on }}
     steps:
     - uses: actions/checkout@v2
 
     - name: Set up Python
       uses: actions/setup-python@v2
       with:
-        python-version: 3.9
+        python-version: '3.9'
 
     - name: Install dependencies
       run: |
         python -m pip install --upgrade pip
-        pip install torch==2.2.2
-        pip install -r requirements.txt
-        pip install -r dev-requirements.txt
-
-    - name: Install package
-      run: |
-        pip install .
-
-    - name: Run tests
-      run: |
-        pytest test --verbose -s -x
-
-  test-cuda-2-3-rc:
-    runs-on: 4-core-ubuntu-gpu-t4
-    steps:
-    - uses: actions/checkout@v2
-
-    - name: Set up Python
-      uses: actions/setup-python@v2
-      with:
-        python-version: 3.9
-
-    - name: Install dependencies
-      run: |
-        python -m pip install --upgrade pip
-        pip install --pre torch --index-url https://download.pytorch.org/whl/nightly/cu121
-        pip install -r requirements.txt
-        pip install -r dev-requirements.txt
-
-  test-cuda-nightly:
-    runs-on: 4-core-ubuntu-gpu-t4
-    steps:
-    - uses: actions/checkout@v2
-
-    - name: Set up Python
-      uses: actions/setup-python@v2
-      with:
-        python-version: 3.9
-
-    - name: Install dependencies
-      run: |
-        python -m pip install --upgrade pip
-        pip install --pre torch --index-url https://download.pytorch.org/whl/nightly/cu121
-        pip install -r requirements.txt
-        pip install -r dev-requirements.txt
-        
-
-    - name: Install package
-      run: |
-        pip install .
-
-    - name: Run tests
-      run: |
-        pytest test --verbose -s -x
-
-  test-cpu:
-    runs-on: 32-core-ubuntu
-    steps:
-    - uses: actions/checkout@v2
-
-    - name: Set up Python
-      uses: actions/setup-python@v2
-      with:
-        python-version: 3.9
-
-    - name: Install dependencies
-      run: |
-        python -m pip install --upgrade pip
-        pip install torch --index-url https://download.pytorch.org/whl/cpu
-        pip install -r requirements.txt
-        pip install -r dev-requirements.txt
-        
-
-    - name: Install package
-      run: |
-        pip install .
-
-    - name: Run tests
-      run: |
-        pytest test --verbose -s -x
-
-  test-nightly-cpu:
-    runs-on: 32-core-ubuntu
-    steps:
-    - uses: actions/checkout@v2
-
-    - name: Set up Python
-      uses: actions/setup-python@v2
-      with:
-        python-version: 3.9
-
-    - name: Install dependencies
-      run: |
-        python -m pip install --upgrade pip
-        pip install --pre torch --index-url https://download.pytorch.org/whl/nightly/cpu
+        pip install ${{ matrix.torch-spec }}
         pip install -r requirements.txt
         pip install -r dev-requirements.txt