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

Pass JAXLIB_* env variables to docker container

Browse Source 
PiperOrigin-RevId: 730922129
This commit is contained in:
Nitin Srinivasan 2025-02-25 09:44:33 -08:00 committed by jax authors
parent ced28167e8
commit 8ac2969759
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
ci/utilities/run_docker_container.sh
View File

@ -52,10 +52,10 @@ if ! docker container inspect jax >/dev/null 2>&1 ; then
JAXCI_DOCKER_ARGS="$JAXCI_DOCKER_ARGS -e GCE_METADATA_HOST=$IP_ADDR"
fi
# Create a temporary file to pass any user defined JAXCI_ / JAX_ variables to
# the container.
# Create a temporary file to pass any user defined JAXCI_ / JAX_ / JAXLIB_
# variables to the container.
JAXCI_TEMP_ENVFILE_DIR=$(mktemp)
env | grep -e "JAXCI_" -e "JAX_" > "$JAXCI_TEMP_ENVFILE_DIR"
env | grep -e "JAXCI_" -e "JAX_" -e "JAXLIB_" > "$JAXCI_TEMP_ENVFILE_DIR"
# On Windows, convert MSYS Linux-like paths to Windows paths.
if [[ "$(uname -s)" =~ "MSYS_NT" ]]; then