[actions] split formatting and unit tests into two jobs (deepspeedai#1339)

jeffra · web-flow · commit 600db0918904 · 2021-09-02T08:46:20.000-07:00
diff --git a/.github/workflows/formatting.yml b/.github/workflows/formatting.yml
@@ -0,0 +1,34 @@
+name: Formatting
+
+on:
+  push:
+    branches:
+      'staging**'
+  pull_request:
+    branches:
+      '**'
+
+jobs:
+
+  # formatting and basic install on cpu-only machine
+  formatting:
+    runs-on: ubuntu-20.04
+
+    steps:
+      - uses: actions/checkout@v2
+
+      - name: environment
+        run: |
+          which python
+          python --version
+          pip install torch==1.9.0+cpu -f https://download.pytorch.org/whl/torch_stable.html
+          python -c "import torch; print('torch:', torch.__version__, torch)"
+
+      - name: Install deepspeed
+        run: |
+          pip install .[dev]
+          ds_report
+
+      - name: Formatting checks
+        run: |
+           pre-commit run --all-files
diff --git a/.github/workflows/main.yml b/.github/workflows/main.yml
@@ -1,29 +1,23 @@
-# This is a basic workflow to help you get started with Actions
-
 name: Build
 
-# Controls when the action will run.
 on:
   push:
+    branches:
+      'staging**'
     paths-ignore:
     - 'docs/**'
   pull_request:
     paths-ignore:
     - 'docs/**'
 
-# A workflow run is made up of one or more jobs that can run sequentially or in parallel
 jobs:
-  # This workflow contains a single job called "build"
-  build:
-    # The type of runner that the job will run on
+  # unit tests running on nvidia gpus
+  unit-tests:
     runs-on: [self-hosted, nvidia]
 
-    # Steps represent a sequence of tasks that will be executed as part of the job
     steps:
-      # Checks-out your repository under $GITHUB_WORKSPACE, so your job can access it
       - uses: actions/checkout@v2
 
-      # Runs a single command using the runners shell
       - name: environment
         run: |
           nvidia-smi
@@ -34,17 +28,11 @@ jobs:
           python -c "import torch; print('torch:', torch.__version__, torch)"
           python -c "import torch; print('CUDA available:', torch.cuda.is_available())"
 
-      # Runs a set of commands using the runners shell
       - name: Install deepspeed
         run: |
           pip install .[dev]
           ds_report
 
-      - name: Formatting checks
-        run: |
-           pre-commit run --all-files
-
-      # Runs a set of commands using the runners shell
       - name: Unit tests
         run: |
           if [[ -d ./torch-extensions ]]; then rm -rf ./torch-extensions; fi
