Skip to content

Commit ee75188

Browse files
authored
[amd] update rocm to 7.0 (#499)
1 parent a409d72 commit ee75188

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

tools/rocm_utils.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,11 +3,14 @@
33
import subprocess
44

55
# defines the default ROCM version to compile against
6-
DEFAULT_ROCM_VERSION = "6.4"
6+
DEFAULT_ROCM_VERSION = "7.0"
77
ROCM_VERSION_MAP = {
88
"6.4": {
99
"pytorch_url": "rocm6.4",
1010
},
11+
"7.0": {
12+
"pytorch_url": "rocm7.0",
13+
},
1114
}
1215

1316

0 commit comments

Comments
 (0)