Skip to content

Commit 0528c9d

Browse files
authored
Remove TPU_LIBRARY_PATH from legacy TPU CI (#7968)
1 parent c201ec1 commit 0528c9d

File tree

1 file changed

+0
-2
lines changed

1 file changed

+0
-2
lines changed

test/tpu/xla_test_job.yaml

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -46,8 +46,6 @@ spec:
4646
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
4747
4848
cd /src/pytorch/xla
49-
# TODO: pallas test requires JAX, now we need to explicitly set TPU_LIBRARY_PATH for JAX, need a permanent fix.
50-
TPU_LIBRARY_PATH=/usr/local/lib/python3.10/site-packages/torch_xla/lib/libtpu.so test/tpu/run_tests.sh
5149
volumeMounts:
5250
- mountPath: /dev/shm
5351
name: dshm

0 commit comments

Comments
 (0)