Fix ub24 dockerfile (#2697)

ROCm · Oct 1, 2024 · ca988e6 · ca988e6
1 parent 863a13f
commit ca988e6
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 4 deletions.
diff --git a/tensorflow/tools/tf_sig_build_dockerfiles/devel.packages.rocm.txt b/tensorflow/tools/tf_sig_build_dockerfiles/devel.packages.rocm.txt
@@ -37,7 +37,6 @@ libssl-dev
 libtool
 libzmq3-dev
 lld-17
-mlocate
 moreutils
 openjdk-11-jdk
 openjdk-11-jre-headless

diff --git a/tensorflow/tools/tf_sig_build_dockerfiles/setup.build-python_ubuntu.sh b/tensorflow/tools/tf_sig_build_dockerfiles/setup.build-python_ubuntu.sh
diff --git a/tensorflow/tools/tf_sig_build_dockerfiles/setup.rocm.sh b/tensorflow/tools/tf_sig_build_dockerfiles/setup.rocm.sh
@@ -31,11 +31,11 @@ if [[ -n $1 ]]; then
     ROCM_VERSION=$1
 fi
 if [[ -n $2 ]]; then
-    if [[ "$2" == "noble" ]] || [[ "$2" == "focal" ]] || [[ "$2" == "jammy" ]] || [[ "$2" == "el7" ]] || [[ "$2" == "el8" ]]; then
+    if [[ "$2" == "focal" ]] || [[ "$2" == "jammy" ]] || [[ "$2" == "noble" ]] || [[ "$2" == "el7" ]] || [[ "$2" == "el8" ]]; then
         DISTRO=$2
     else
         echo "Distro not supported"
-        echo "Supported distros are:\n noble\n focal\n jammy\n el7\n el8"
+        echo "Supported distros are:\n focal\n jammy\n noble\n el7\n el8"
 	exit 1
     fi
 fi
@@ -49,7 +49,7 @@ else
         ROCM_VERS=$ROCM_VERSION
 fi
 
-if [[ "$DISTRO" == "noble" ]] || [[ "$DISTRO" == "focal" ]] || [[ "$DISTRO" == "jammy" ]]; then
+if [[ "$DISTRO" == "focal" ]] || [[ "$DISTRO" == "jammy" ]] || [[ "$DISTRO" == "noble" ]]; then
     ROCM_DEB_REPO_HOME=https://repo.radeon.com/rocm/apt/
     AMDGPU_DEB_REPO_HOME=https://repo.radeon.com/amdgpu/
     ROCM_BUILD_NAME=${DISTRO}