From 34a7fbdfec6fcf6b33da077ffc5ed5bf9d9c332c Mon Sep 17 00:00:00 2001
From: epwalsh <petew@allenai.org>
Date: Tue, 2 Apr 2024 15:52:41 -0700
Subject: [PATCH] break up tests more

---
 .github/workflows/main.yml | 14 +++++++++++---
 1 file changed, 11 insertions(+), 3 deletions(-)

diff --git a/.github/workflows/main.yml b/.github/workflows/main.yml
index 6f86269c..66ab5560 100644
--- a/.github/workflows/main.yml
+++ b/.github/workflows/main.yml
@@ -24,7 +24,9 @@ env:
 jobs:
   checks:
     name: ${{ matrix.task.name }}
-    runs-on: [macos-13]  # TODO: change to ubuntu-latest once repo is public (will have more RAM then)
+    # TODO: change to 'ubuntu-latest' once repo is public (will have more RAM then), and update the torch
+    # install command in the setup-venv action.
+    runs-on: [macos-13]
     timeout-minutes: 5
     strategy:
       fail-fast: false
@@ -34,9 +36,15 @@ jobs:
           - name: Lint
             run: make lint-check
 
-          - name: Test core
+          - name: Test
             run: |
-              pytest -v --color=yes --durations=3 src/test/ --ignore-glob='src/test/distributed/fsdp/*'
+              pytest -v --color=yes --durations=3 src/test/ \
+                --ignore-glob='src/test/distributed/fsdp/*' \
+                --ignore-glob='src/test/distributed/sharded_flat_*'
+
+          - name: Test tensors
+            run: |
+              pytest -v --color=yes --durations=3 src/test/distributed/sharded_flat_*
 
           - name: Test FSDP
             run: |