mirror of
https://github.com/ROCm/jax.git
synced 2025-04-16 11:56:07 +00:00
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:
parent
3b97ff2b6d
commit
719cae5849
4
.github/workflows/cloud-tpu-ci-nightly.yml
vendored
4
.github/workflows/cloud-tpu-ci-nightly.yml
vendored
@ -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
|
||||
|
Loading…
x
Reference in New Issue
Block a user