From e941cb1ff71ae619e489096dacf733b659977cb2 Mon Sep 17 00:00:00 2001 From: Shuwen-Fang Date: Mon, 1 Jun 2026 20:09:54 +0000 Subject: [PATCH] t --- .github/workflows/run_tests_against_package.yml | 16 +--------------- 1 file changed, 1 insertion(+), 15 deletions(-) diff --git a/.github/workflows/run_tests_against_package.yml b/.github/workflows/run_tests_against_package.yml index 5c44e268bc..f905bab424 100644 --- a/.github/workflows/run_tests_against_package.yml +++ b/.github/workflows/run_tests_against_package.yml @@ -153,23 +153,9 @@ jobs: export MAXTEXT_ASSETS_ROOT=$(pwd)/src/maxtext/assets export MAXTEXT_TEST_ASSETS_ROOT=$(pwd)/tests/assets export MAXTEXT_PKG_DIR=$(pwd)/src/maxtext - # omit this libtpu init args for gpu tests if [ "${INPUTS_DEVICE_TYPE}" != "cuda12" ]; then export LIBTPU_INIT_ARGS='--xla_tpu_scoped_vmem_limit_kib=65536' - else - # For cuda12, explicitly point to the pip-installed CUDA libraries - # to avoid conflicts with system-level installations on the runner. - # Dynamically discover the 'nvidia' folder and prepend all its sub-library - # directories (including nccl, cublas, cudnn) to LD_LIBRARY_PATH to prevent - # JAX from partially loading incompatible system-level CUDA libraries. - NVIDIA_DIR=$(find .venv/lib/ -maxdepth 3 -name "nvidia" -type d 2>/dev/null | head -n 1) - if [ -n "${NVIDIA_DIR}" ]; then - for dir in "${NVIDIA_DIR}"/*; do - if [ -d "$dir/lib" ]; then - export LD_LIBRARY_PATH=$(pwd)/$dir/lib:${LD_LIBRARY_PATH} - fi - done - fi + echo "DEBUG: Set LIBTPU_INIT_ARGS for non-cuda12" fi if [ "${INPUTS_TOTAL_WORKERS}" -gt 1 ]; then $PYTHON_EXE -m pip install --quiet pytest-split pytest-xdist