diff --git a/.circleci/docker/common/install_rocm.sh b/.circleci/docker/common/install_rocm.sh index 7dec36c57214db..6347f36420cd7d 100644 --- a/.circleci/docker/common/install_rocm.sh +++ b/.circleci/docker/common/install_rocm.sh @@ -35,7 +35,7 @@ ver() { } # Map ROCm version to AMDGPU version -declare -A AMDGPU_VERSIONS=( ["4.5.2"]="21.40.2" ["5.0"]="21.50" ["5.2"]="22.20" ) +declare -A AMDGPU_VERSIONS=( ["4.5.2"]="21.40.2" ["5.0"]="21.50" ["5.2"]="22.20" ["5.2.1"]="22.20.1" ) install_ubuntu() { apt-get update @@ -117,7 +117,7 @@ install_centos() { echo "gpgkey=http://repo.radeon.com/rocm/rocm.gpg.key" >> /etc/yum.repos.d/amdgpu.repo fi - local rocm_baseurl="http://repo.radeon.com/rocm/yum/${ROCM_VERSION}" + local rocm_baseurl="http://repo.radeon.com/rocm/yum/${ROCM_VERSION}/main" echo "[ROCm]" > /etc/yum.repos.d/rocm.repo echo "name=ROCm" >> /etc/yum.repos.d/rocm.repo echo "baseurl=${rocm_baseurl}" >> /etc/yum.repos.d/rocm.repo