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

Merge pull request #24523 from ROCm:ci_build_fix_2

Browse Source 
PiperOrigin-RevId: 689787724
This commit is contained in:
jax authors 2024-10-25 07:57:33 -07:00
commit 69b7129a87
1 changed files with 8 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
build/rocm/ci_build.sh
View File

@ -53,7 +53,7 @@ CUSTOM_INSTALL=""
JAX_USE_CLANG=""
POSITIONAL_ARGS=()
RUNTIME_FLAG=1
RUNTIME_FLAG=0
while [[ $# -gt 0 ]]; do
case $1 in
@ -113,11 +113,13 @@ function upsearch (){
}
# Set up WORKSPACE.
WORKSPACE="${WORKSPACE:-$(upsearch WORKSPACE)}"
BUILD_TAG="${BUILD_TAG:-jax}"
# Determine the docker image name and BUILD_TAG.
DOCKER_IMG_NAME="${BUILD_TAG}.${CONTAINER_TYPE}"
if [ ${RUNTIME_FLAG} -eq 0 ]; then
DOCKER_IMG_NAME="${BUILD_TAG}"
else
WORKSPACE="${WORKSPACE:-$(upsearch WORKSPACE)}"
BUILD_TAG="${BUILD_TAG:-jax}"
DOCKER_IMG_NAME="${BUILD_TAG}.${CONTAINER_TYPE}"
fi
# Under Jenkins matrix build, the build tag may contain characters such as
# commas (,) and equal signs (=), which are not valid inside docker image names.