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

Cloud TPU CI: don't use tcmalloc (temporary workaround for tcmalloc deadlock)

Browse Source
This commit is contained in:
Skye Wanderman-Milne 2023-09-28 16:55:11 -07:00
parent e6f8477efe
commit ef241e506e
1 changed files with 5 additions and 0 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,6 +71,11 @@ 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 \