From 91b483b8aaa54d945f7c6f8b1cd708a4b283ee23 Mon Sep 17 00:00:00 2001
From: Jithun Nair <jithun.nair@amd.com>
Date: Tue, 17 Jan 2023 19:50:28 +0000
Subject: [PATCH] Set ubuntu distribution correctly for ROCm5.3 and above

---
 common/install_rocm.sh | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/common/install_rocm.sh b/common/install_rocm.sh
index 6fca6cfa1..75e7e6e27 100644
--- a/common/install_rocm.sh
+++ b/common/install_rocm.sh
@@ -47,6 +47,10 @@ install_ubuntu() {
         ROCM_REPO="xenial"
     fi
 
+    if [[ $(ver $ROCM_VERSION) -ge $(ver 5.3) ]]; then
+        ROCM_REPO="${UBUNTU_VERSION_NAME}"
+    fi
+
     # Add rocm repository
     wget -qO - http://repo.radeon.com/rocm/rocm.gpg.key | apt-key add -
     local rocm_baseurl="http://repo.radeon.com/rocm/apt/${ROCM_VERSION}"