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

Print the error

Browse Source
This commit is contained in:
Frederic Bastien 2023-04-17 07:49:08 -07:00
parent 8ce19eea4f
commit 30f30dd31d
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
.github/workflows/slurm_job_scripts/run_e2e_t5x_tests.sub vendored
View File

@ -48,7 +48,7 @@ read -r -d '' setup_cmd <<EOF
rm -rf ${E2E_TESTS_WORKSPACE_DIR}/* \
&& mkdir -p ${E2E_TESTS_WORKSPACE_DIR} \
&& mkdir -p ${TFDS_DATA_DIR} \
&& JAXLIB=`wget -O - https://storage.googleapis.com/jax-releases/jaxlib_nightly_cuda_releases.html 2>/dev/null |grep cuda11|grep -oP 'ht.*?whl'|sort|tail -n 1` \
&& JAXLIB=`wget -O - https://storage.googleapis.com/jax-releases/jaxlib_nightly_cuda_releases.html |grep cuda11|grep -oP 'ht.*?whl'|sort|tail -n 1` \
&& echo "JAXLIB=${JAXLIB}" \
&& python3.8 -m pip install git+https://github.com/google/jax \
&& python3.8 -m pip install ${JAXLIB} \