From a9a92dff0ba10a848720a5e07878e9d2fc47d750 Mon Sep 17 00:00:00 2001
From: Billy Moses <gh@wsmoses.com>
Date: Tue, 3 Sep 2024 20:34:23 +0000
Subject: [PATCH] Now with tpu CI?

---
 .github/workflows/build.yml | 12 ++++++++----
 1 file changed, 8 insertions(+), 4 deletions(-)

diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml
index 28fd0023..db52cd52 100644
--- a/.github/workflows/build.yml
+++ b/.github/workflows/build.yml
@@ -14,7 +14,7 @@ jobs:
     strategy:
       fail-fast: false
       matrix:
-        os: [openstack22]
+        os: [openstack22, tpu]
         python: ["3.12"]
     timeout-minutes: 500
     steps:
@@ -22,9 +22,7 @@ jobs:
         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"