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

Switch ROCm CI scritps to use ROCm 4.0

Browse Source
This commit is contained in:
Deven Desai 2020-12-22 16:13:49 +00:00
parent 14ff3b1566
commit 3d90190923
4 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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.9.0
ROCM_INSTALL_DIR=/opt/rocm-4.0.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.9.0
ROCM_INSTALL_DIR=/opt/rocm-4.0.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.9.0
ROCM_INSTALL_DIR=/opt/rocm-4.0.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.9.0
ROCM_INSTALL_DIR=/opt/rocm-4.0.0
if [[ -n $1 ]]; then
ROCM_INSTALL_DIR=$1
fi