Small Cloud TPU CI debugging improvements

1. Use `strings` to print the libtpu build date, which is useful if
   the libtpu version is stale and causes an API version mismatch
   error.

2. Set the default shell to add bash -x command, so all commands are
   printed inline. This makes lining up the output with the command
   easier.
This commit is contained in:
Skye Wanderman-Milne 2023-08-14 10:37:26 -07:00
parent 3b97ff2b6d
commit 719cae5849

View File

@ -17,6 +17,9 @@ jobs:
name: "TPU test (${{ matrix.jaxlib-version }}, ${{ matrix.tpu-type }})"
runs-on: ["self-hosted", "tpu", "${{ matrix.tpu-type }}"]
timeout-minutes: 60
defaults:
run:
shell: bash -ex {0}
steps:
# https://opensource.google/documentation/reference/github/services#actions
# mandates using a specific commit for non-Google actions. We use
@ -49,6 +52,7 @@ jobs:
python3 -c 'import sys; print("python version:", sys.version)'
python3 -c 'import jax; print("jax version:", jax.__version__)'
python3 -c 'import jaxlib; print("jaxlib version:", jaxlib.__version__)'
strings $HOME/.local/lib/python3.10/site-packages/libtpu/libtpu.so | grep 'Built on'
python3 -c 'import jax; print("libtpu version:",
jax.lib.xla_bridge.get_backend().platform_version)'
- name: Run tests