From de8086e14ae3152906e1137c212d2f7bb8ea463a Mon Sep 17 00:00:00 2001
From: jayfurmanek <Jason.Furmanek@amd.com>
Date: Fri, 30 Jun 2023 12:05:35 -0500
Subject: [PATCH] Update to ROCm5.6 (#2145)

---
 build_rocm_python3                                            | 2 +-
 tensorflow/tools/ci_build/Dockerfile.rocm                     | 4 ++--
 tensorflow/tools/ci_build/linux/rocm/run_gpu_multi.sh         | 2 +-
 tensorflow/tools/ci_build/linux/rocm/run_gpu_single.sh        | 2 +-
 tensorflow/tools/tf_sig_build_dockerfiles/Dockerfile.rocm.rt  | 2 +-
 .../tools/tf_sig_build_dockerfiles/Dockerfile.rocm.ub20       | 2 +-
 tensorflow/tools/tf_sig_build_dockerfiles/README.md           | 2 +-
 7 files changed, 8 insertions(+), 8 deletions(-)

diff --git a/build_rocm_python3 b/build_rocm_python3
index 2e970ed4b17d23..022096eab81947 100755
--- a/build_rocm_python3
+++ b/build_rocm_python3
@@ -26,7 +26,7 @@ done
 shift "$((OPTIND-1))"
 
 # First positional argument (if any) specifies the ROCM_INSTALL_DIR
-ROCM_INSTALL_DIR=/opt/rocm-5.5.0
+ROCM_INSTALL_DIR=/opt/rocm-5.6.0
 if [[ -n $1 ]]; then
     ROCM_INSTALL_DIR=$1
 fi
diff --git a/tensorflow/tools/ci_build/Dockerfile.rocm b/tensorflow/tools/ci_build/Dockerfile.rocm
index 3c26a6ceac4e5a..fe420591c0e0ac 100644
--- a/tensorflow/tools/ci_build/Dockerfile.rocm
+++ b/tensorflow/tools/ci_build/Dockerfile.rocm
@@ -3,10 +3,10 @@
 FROM ubuntu:focal
 MAINTAINER Jeff Poznanovic <jeffrey.poznanovic@amd.com>
 
-ARG ROCM_DEB_REPO=https://repo.radeon.com/rocm/apt/5.5/
+ARG ROCM_DEB_REPO=https://repo.radeon.com/rocm/apt/5.6/
 ARG ROCM_BUILD_NAME=ubuntu
 ARG ROCM_BUILD_NUM=main
-ARG ROCM_PATH=/opt/rocm-5.5.0
+ARG ROCM_PATH=/opt/rocm-5.6.0
 ARG GPU_DEVICE_TARGETS="gfx900 gfx906 gfx908 gfx90a gfx1030"
 
 ARG DEBIAN_FRONTEND=noninteractive
diff --git a/tensorflow/tools/ci_build/linux/rocm/run_gpu_multi.sh b/tensorflow/tools/ci_build/linux/rocm/run_gpu_multi.sh
index cf1eb6edaae6ca..c04de95e30d12d 100755
--- a/tensorflow/tools/ci_build/linux/rocm/run_gpu_multi.sh
+++ b/tensorflow/tools/ci_build/linux/rocm/run_gpu_multi.sh
@@ -25,7 +25,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-5.5.0
+ROCM_INSTALL_DIR=/opt/rocm-5.6.0
 if [[ -n $1 ]]; then
     ROCM_INSTALL_DIR=$1
 fi
diff --git a/tensorflow/tools/ci_build/linux/rocm/run_gpu_single.sh b/tensorflow/tools/ci_build/linux/rocm/run_gpu_single.sh
index 14837f800837c4..77a327f6d9875a 100755
--- a/tensorflow/tools/ci_build/linux/rocm/run_gpu_single.sh
+++ b/tensorflow/tools/ci_build/linux/rocm/run_gpu_single.sh
@@ -27,7 +27,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-5.5.0
+ROCM_INSTALL_DIR=/opt/rocm-5.6.0
 if [[ -n $1 ]]; then
     ROCM_INSTALL_DIR=$1
 fi
diff --git a/tensorflow/tools/tf_sig_build_dockerfiles/Dockerfile.rocm.rt b/tensorflow/tools/tf_sig_build_dockerfiles/Dockerfile.rocm.rt
index 42e727aa403be5..970b62fa33b2f8 100644
--- a/tensorflow/tools/tf_sig_build_dockerfiles/Dockerfile.rocm.rt
+++ b/tensorflow/tools/tf_sig_build_dockerfiles/Dockerfile.rocm.rt
@@ -12,7 +12,7 @@ RUN /setup.packages.sh /runtime.packages.txt
 # Install ROCM
 ARG TF_PKGS_DIR=tmp/packages
 ARG TENSORFLOW_PACKAGE=tf_nightly_rocm
-ARG ROCM_VERSION=5.5.0
+ARG ROCM_VERSION=5.6.0
 ARG CUSTOM_INSTALL
 ARG ROCM_PATH=/opt/rocm-${ROCM_VERSION}
 ENV ROCM_PATH=${ROCM_PATH}
diff --git a/tensorflow/tools/tf_sig_build_dockerfiles/Dockerfile.rocm.ub20 b/tensorflow/tools/tf_sig_build_dockerfiles/Dockerfile.rocm.ub20
index 38da221214c1ab..9c5e9a748ff06e 100644
--- a/tensorflow/tools/tf_sig_build_dockerfiles/Dockerfile.rocm.ub20
+++ b/tensorflow/tools/tf_sig_build_dockerfiles/Dockerfile.rocm.ub20
@@ -11,7 +11,7 @@ COPY builder.packages.txt builder.packages.txt
 RUN /setup.packages.sh /builder.packages.txt
 
 # Install ROCM
-ARG ROCM_VERSION=5.5.0
+ARG ROCM_VERSION=5.6.0
 ARG CUSTOM_INSTALL
 ARG ROCM_PATH=/opt/rocm-${ROCM_VERSION}
 ENV ROCM_PATH=${ROCM_PATH}
diff --git a/tensorflow/tools/tf_sig_build_dockerfiles/README.md b/tensorflow/tools/tf_sig_build_dockerfiles/README.md
index 67c9ff0d6159e9..81a740711663c6 100644
--- a/tensorflow/tools/tf_sig_build_dockerfiles/README.md
+++ b/tensorflow/tools/tf_sig_build_dockerfiles/README.md
@@ -50,7 +50,7 @@ DOCKER_BUILDKIT=1 docker build \
 For ROCM
 ```
 DOCKER_BUILDKIT=1 docker build -f Dockerfile.rocm \
-  --build-arg ROCM_VERSION=5.5.0 --build-arg PYTHON_VERSION=3.9 -t my-tf-devel . 
+  --build-arg ROCM_VERSION=5.6.0 --build-arg PYTHON_VERSION=3.9 -t my-tf-devel . 
 ```
 It will take a long time to build devtoolset and install packages. After
 it's done, you can use the commands below to test your changes. Just replace