Skip to content

Commit e503deb

Browse files
WBobbyyanyao-wang
authored andcommitted
Add ROCm5.0/AMDGPU support (#904)
Signed-off-by: Wang, Yanyao <[email protected]> Co-authored-by: Wang, Yanyao <[email protected]>
1 parent 5ee9bed commit e503deb

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
@@ -27,7 +27,7 @@ ver() {
2727
}
2828

2929
# Map ROCm version to AMDGPU version
30-
declare -A AMDGPU_VERSIONS=( ["4.5.2"]="21.40.2" )
30+
declare -A AMDGPU_VERSIONS=( ["4.5.2"]="21.40.2" ["5.0"]="21.50" )
3131

3232
install_ubuntu() {
3333
apt-get update

0 commit comments

Comments
 (0)