add libdrm-amdgpu-dev for focal image with empty hipDeviceGetName()

tensorflow/tools/ci_build/Dockerfile.rocm

@@ -23,6 +23,7 @@ RUN bin/bash -c 'if [[ $ROCM_DEB_REPO == https://repo.radeon.com/rocm/*  ]] ; th
     else \
       echo "deb [arch=amd64 trusted=yes] $ROCM_DEB_REPO $ROCM_BUILD_NAME $ROCM_BUILD_NUM" > /etc/apt/sources.list.d/rocm.list ; \
     fi'
+RUN echo "deb [arch=amd64] https://repo.radeon.com/amdgpu/6.0/ubuntu focal $ROCM_BUILD_NUM" > /etc/apt/sources.list.d/amdgpu.list
 
 # Install misc pkgs
 RUN apt-get update --allow-insecure-repositories && DEBIAN_FRONTEND=noninteractive apt-get install -y \
@@ -78,7 +79,7 @@ RUN MIOPENKERNELS=$( \
     ) && \
     apt-get update --allow-insecure-repositories && \
     DEBIAN_FRONTEND=noninteractive apt-get install -y --allow-unauthenticated \
-    rocm-dev rocm-libs ${MIOPENKERNELS} && \
+    libdrm-amdgpu-dev rocm-dev rocm-libs ${MIOPENKERNELS} && \
     apt-get clean && \
     rm -rf /var/lib/apt/lists/*
 

tensorflow/tools/tf_sig_build_dockerfiles/devel.packages.rocm.txt

@@ -1,4 +1,5 @@
 # All required ROCM packages
+libdrm-amdgpu-dev
 rocm-libs
 rocm-dev
 rocm-ml-sdk

tensorflow/tools/tf_sig_build_dockerfiles/setup.rocm.sh

@@ -51,6 +51,7 @@ if [[ $ROCM_DEB_REPO == https://repo.radeon.com/rocm/*  ]] ; then \
 else
     bash "/${CUSTOM_INSTALL}"
 fi
+echo "deb [arch=amd64] https://repo.radeon.com/amdgpu/6.0/ubuntu focal $ROCM_BUILD_NUM" > /etc/apt/sources.list.d/amdgpu.list
 
 GPU_DEVICE_TARGETS=${GPU_DEVICE_TARGETS:-"gfx900 gfx906 gfx908 gfx90a gfx940 gfx941 gfx942 gfx1030 gfx1100"}