mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-18 21:06:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

Remove the LD_PRELOAD workaround introduced in ef241e506e

Browse Source 
libtpu has been updated to hopefully avoid the deadlock.
This commit is contained in:
Skye Wanderman-Milne 2023-10-02 11:30:20 -07:00
parent 228e1925b2
commit 24c9e62910
1 changed files with 0 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
.github/workflows/cloud-tpu-ci-nightly.yml vendored
View File

@ -71,11 +71,6 @@ jobs:
env:
JAX_PLATFORMS: tpu,cpu
PY_COLORS: 1
# TODO(b/302562327): TPU VMs have an ancient tcmalloc set in their
# LD_PRELOAD by default. This has been causing deadlocks due to its
# use of libunwind. Use the system allocator in the meantime to
# unblock testing until we develop a better fix.
LD_PRELOAD:
run: |
# Run single-accelerator tests in parallel
JAX_ENABLE_TPU_XDIST=true python3 -m pytest -n=4 --tb=short \