From 21c1c63eab2125490b127c519743fd5944d2ae4d Mon Sep 17 00:00:00 2001 From: Will Cromar Date: Tue, 12 Mar 2024 15:13:21 -0700 Subject: [PATCH] Unbundle libtpu in CI (#6727) --- .github/workflows/tpu_ci.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/workflows/tpu_ci.yml b/.github/workflows/tpu_ci.yml index e031489f839..12494333552 100644 --- a/.github/workflows/tpu_ci.yml +++ b/.github/workflows/tpu_ci.yml @@ -23,7 +23,6 @@ jobs: - name: Run PyTorch/XLA Setup env: BAZEL_VERBOSE: 1 - BUNDLE_LIBTPU: 1 TPUVM_MODE: 1 run: | cd pytorch/xla @@ -37,5 +36,6 @@ jobs: pip install fsspec pip install rich pip install torch_xla[pallas] -f https://storage.googleapis.com/jax-releases/jax_nightly_releases.html -f https://storage.googleapis.com/jax-releases/jaxlib_nightly_releases.html + pip install torch_xla[tpuvm] cd pytorch/xla test/tpu/run_tests.sh