run levanter tests in Haliax

.github/workflows/run_pre_commit.yaml

@@ -1,6 +1,6 @@
 name: Pre-Commit
 
-on: [push]
+on: [push, pull_request]
 
 jobs:
   build:

.github/workflows/run_quick_levanter_tests.yaml

@@ -0,0 +1,36 @@
+name: Run Tests
+
+on: [push, pull_request]
+
+jobs:
+  build:
+
+    runs-on: ubuntu-latest
+
+    steps:
+      - uses: actions/checkout@v3
+      - name: Set up Python 3.10.11
+        uses: actions/setup-python@v4
+        with:
+          python-version: 3.10.11
+      - name: Install dependencies
+        run: |
+          python -m pip install --upgrade pip
+          pip install flake8 pytest
+          pip install --upgrade "jax[cpu]==0.4.25" "jaxlib[cpu]==0.4.25"
+
+      - name: Install Levanter from source
+        run: |
+          git clone "git+https://github.com/stanford-crfm/levanter.git"
+          cd levanter
+          pip install .
+
+      - name: Install Haliax on top
+        run: |
+          # install levanter main
+          cd ..
+          pip install .[dev]
+      - name: Test levanter with pytest
+        run: |
+          cd levanter
+          XLA_FLAGS=--xla_force_host_platform_device_count=8 PYTHONPATH=tests:src:../src pytest tests -m "not entry and not slow"

.github/workflows/run_tests.yaml

@@ -1,21 +1,18 @@
 name: Run Tests
 
-on: [push]
+on: [push, pull_request]
 
 jobs:
   build:
 
     runs-on: ubuntu-latest
-    strategy:
-      matrix:
-        python-version: ["3.10.11"]
 
     steps:
       - uses: actions/checkout@v3
-      - name: Set up Python ${{ matrix.python-version }}
+      - name: Set up Python 3.10.11
         uses: actions/setup-python@v4
         with:
-          python-version: ${{ matrix.python-version }}
+          python-version: 3.10.11
       - name: Install dependencies
         run: |
           python -m pip install --upgrade pip