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

Pass in "JAX_" variables to the docker container

Browse Source 
PiperOrigin-RevId: 730689480
This commit is contained in:
Nitin Srinivasan 2025-02-24 19:31:29 -08:00 committed by jax authors
parent b707f0bdbb
commit e4d63dff3d
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_ variables to the
# container.
# Create a temporary file to pass any user defined JAXCI_ / JAX_ variables to
# the container.
JAXCI_TEMP_ENVFILE_DIR=$(mktemp)
env | grep "JAXCI_" > "$JAXCI_TEMP_ENVFILE_DIR"
env | grep -e "JAXCI_" -e "JAX_" > "$JAXCI_TEMP_ENVFILE_DIR"
# On Windows, convert MSYS Linux-like paths to Windows paths.
if [[ "$(uname -s)" =~ "MSYS_NT" ]]; then