There was an error while loading. Please reload this page.
TPU_LIBRARY_PATH
1 parent c201ec1 commit 0528c9dCopy full SHA for 0528c9d
test/tpu/xla_test_job.yaml
@@ -46,8 +46,6 @@ spec:
46
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
47
48
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
51
volumeMounts:
52
- mountPath: /dev/shm
53
name: dshm
0 commit comments