Skip to content

Commit cbdfcac

Browse files
WBobbyyanyao-wang
authored andcommitted
Add amdgpu version support for ROCm5.1 (#980)
Co-authored-by: Wang, Yanyao <[email protected]>
1 parent 6c589ac commit cbdfcac

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

.circleci/docker/common/install_rocm.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -29,7 +29,7 @@ ver() {
2929
}
3030

3131
# Map ROCm version to AMDGPU version
32-
declare -A AMDGPU_VERSIONS=( ["4.5.2"]="21.40.2" ["5.0"]="21.50" ["5.0.1"]="21.50.1" )
32+
declare -A AMDGPU_VERSIONS=( ["4.5.2"]="21.40.2" ["5.0"]="21.50" ["5.0.1"]="21.50.1" ["5.1"]="22.10" )
3333

3434
install_ubuntu() {
3535
apt-get update

0 commit comments

Comments
 (0)