Unbundle libtpu in CI (#6727)

pytorch · Mar 12, 2024 · 21c1c63 · 21c1c63
1 parent bf60b62
commit 21c1c63
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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