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

Try to fix strange error where DATE end up being empty.\n\n. All is working locally.

Browse Source
This commit is contained in:
Frederic Bastien 2023-04-11 12:06:05 -07:00
parent 466c9a282d
commit 951f5c5ad1
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -49,10 +49,10 @@ rm -rf ${E2E_TESTS_WORKSPACE_DIR}/* \
&& mkdir -p ${E2E_TESTS_WORKSPACE_DIR} \
&& mkdir -p ${TFDS_DATA_DIR} \
&& DATE=`date --date="yesterday" +%Y%m%d` \
&& echo "WHEEL_DATE=$DATE" \
&& JAXLIB=`wget -O - https://storage.googleapis.com/jax-releases/jaxlib_nightly_cuda_releases.html 2>/dev/null |grep $DATE|grep cuda11|grep -oP 'ht.*?whl'` \
&& echo "WHEEL_DATE=${DATE}" \
&& JAXLIB=`wget -O - https://storage.googleapis.com/jax-releases/jaxlib_nightly_cuda_releases.html 2>/dev/null |grep ${DATE}|grep cuda11|grep -oP 'ht.*?whl'` \
&& python3.8 -m pip install git+https://github.com/google/jax \
&& python3.8 -m pip install $JAXLIB \
&& python3.8 -m pip install ${JAXLIB} \
&& git clone https://github.com/google-research/t5x.git ${T5X_DIR} \
&& python3.8 -m pip uninstall -y cudf dask-cudf \
&& python3.8 -m pip install ${T5X_DIR} \