Skip to content

Commit ee4d862

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 beadc08 commit ee4d862

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
@@ -30,7 +30,7 @@ ver() {
3030
}
3131

3232
# Map ROCm version to AMDGPU version
33-
declare -A AMDGPU_VERSIONS=( ["4.5.2"]="21.40.2" )
33+
declare -A AMDGPU_VERSIONS=( ["4.5.2"]="21.40.2" ["5.0"]="21.50" )
3434

3535
install_ubuntu() {
3636
apt-get update

0 commit comments

Comments
 (0)