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 1/3] 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"

From dc41b4d320f6ce2064a5097f2e44682abc1ed4ac Mon Sep 17 00:00:00 2001
From: Billy Moses <gh@wsmoses.com>
Date: Tue, 3 Sep 2024 20:38:19 +0000
Subject: [PATCH 2/3] fix

---
 .github/workflows/build.yml | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml
index db52cd52..cf9b772d 100644
--- a/.github/workflows/build.yml
+++ b/.github/workflows/build.yml
@@ -60,5 +60,4 @@ jobs:
         uses: actions/upload-artifact@v3
         with:
           name: Wheel-${{ matrix.os }}
-          path: "*.whl"
-          path: "*.log"
+          path: "*.{whl,log}"

From 5d782abcddb8683da1efa38beaf116b99d5bab4c Mon Sep 17 00:00:00 2001
From: Billy Moses <gh@wsmoses.com>
Date: Tue, 3 Sep 2024 21:14:37 +0000
Subject: [PATCH 3/3] Cache tpu

---
 .github/workflows/build.yml | 7 +++++--
 test/BUILD                  | 1 +
 2 files changed, 6 insertions(+), 2 deletions(-)

diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml
index cf9b772d..dc363baa 100644
--- a/.github/workflows/build.yml
+++ b/.github/workflows/build.yml
@@ -28,8 +28,11 @@ jobs:
         with:
           submodules: recursive
       
-      - run: sudo rm -rf ~/.cache/bazel ~/.cache/bazelisk
-      
+      - run: |
+            if [ "${{matrix.os}}" == "openstack22" ]; then
+              sudo rm -rf ~/.cache/bazel ~/.cache/bazelisk
+            fi
+
       - uses: bazel-contrib/setup-bazel@0.8.5
         name: Set up Bazel
         with:
diff --git a/test/BUILD b/test/BUILD
index 6558078e..5b43a375 100644
--- a/test/BUILD
+++ b/test/BUILD
@@ -108,6 +108,7 @@ py_test(
     ],
     imports = ["."],
     deps = TEST_DEPS,
+    timeout='long'
 )
 
 py_test(