Now with tpu CI?

EnzymeAD · Sep 3, 2024 · a9a92df · a9a92df
1 parent 27ccd5e
commit a9a92df
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml
@@ -14,17 +14,15 @@ jobs:
     strategy:
       fail-fast: false
       matrix:
-        os: [openstack22]
+        os: [openstack22, tpu]
         python: ["3.12"]
     timeout-minutes: 500
     steps:
       - name: add llvm
         run: |
             if [ "$(expr substr $(uname -s) 1 5)" == "Linux" ]; then
               sudo apt-get update
-              sudo apt-get install -y git gcc g++ #python3 python3-dev python3-pip
-              #sudo python3 -m pip install --upgrade lit
-              #- run: python3 -m pip install --user numpy
+              sudo apt-get install -y git gcc g++ 
             fi
       - uses: actions/checkout@v3
         with:
@@ -46,6 +44,11 @@ jobs:
       - name: test
         run: |
           HERMETIC_PYTHON_VERSION=${{ matrix.python }} bazel test --test_output=errors ...
+          HERMETIC_PYTHON_VERSION=${{ matrix.python }} bazel test --cache_test_results=no //test:bench_vs_xla
+          HERMETIC_PYTHON_VERSION=${{ matrix.python }} bazel test --cache_test_results=no //test:llama
+          cat bazel-out/*/testlogs/test/llama/test.log
+          cp bazel-out/*/testlogs/test/llama/test.log llama.log
+
           
       - name: Build Wheel
         run: |
@@ -58,3 +61,4 @@ jobs:
         with:
           name: Wheel-${{ matrix.os }}
           path: "*.whl"
+          path: "*.log"