experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Updating ROCm CI scripts to use ROCm 3.9

Browse Source
This commit is contained in:
Deven Desai 2020-12-04 02:17:37 +00:00
parent 0fd4fc2107
commit 639ad2c43d
4 changed files with 4 additions and 4 deletions
tensorflow/tools/ci_build
linux/rocm
run_cc_core.shrun_csb_tests.shrun_py3_core.sh
xla/linux/rocm
run_py3.sh

2
tensorflow/tools/ci_build/linux/rocm/run_cc_core.sh
View File

@ -28,7 +28,7 @@ echo "Bazel will use ${N_BUILD_JOBS} concurrent build job(s) and ${N_TEST_JOBS}
echo ""
# First positional argument (if any) specifies the ROCM_INSTALL_DIR
ROCM_INSTALL_DIR=/opt/rocm-3.7.0
ROCM_INSTALL_DIR=/opt/rocm-3.9.0
if [[ -n $1 ]]; then
ROCM_INSTALL_DIR=$1
fi

2
tensorflow/tools/ci_build/linux/rocm/run_csb_tests.sh
View File

@ -28,7 +28,7 @@ echo "Bazel will use ${N_BUILD_JOBS} concurrent build job(s) and ${N_TEST_JOBS}
echo ""
# First positional argument (if any) specifies the ROCM_INSTALL_DIR
ROCM_INSTALL_DIR=/opt/rocm-3.7.0
ROCM_INSTALL_DIR=/opt/rocm-3.9.0
if [[ -n $1 ]]; then
ROCM_INSTALL_DIR=$1
fi

2
tensorflow/tools/ci_build/linux/rocm/run_py3_core.sh
View File

@ -28,7 +28,7 @@ echo "Bazel will use ${N_BUILD_JOBS} concurrent build job(s) and ${N_TEST_JOBS}
echo ""
# First positional argument (if any) specifies the ROCM_INSTALL_DIR
ROCM_INSTALL_DIR=/opt/rocm-3.7.0
ROCM_INSTALL_DIR=/opt/rocm-3.9.0
if [[ -n $1 ]]; then
ROCM_INSTALL_DIR=$1
fi

2
tensorflow/tools/ci_build/xla/linux/rocm/run_py3.sh
View File

@ -28,7 +28,7 @@ echo "Bazel will use ${N_BUILD_JOBS} concurrent build job(s) and ${N_TEST_JOBS}
echo ""
# First positional argument (if any) specifies the ROCM_INSTALL_DIR
ROCM_INSTALL_DIR=/opt/rocm-3.7.0
ROCM_INSTALL_DIR=/opt/rocm-3.9.0
if [[ -n $1 ]]; then
ROCM_INSTALL_DIR=$1
fi