Skip to content

Commit 8228167

Browse files
xinyazhangethanwee1
authored andcommitted
AOTriton: add 0.9.2b version built on ROCM 6.5, with gfx950 supported. (#2105)
Also fixes the URL problem, where release page does not always match the version string in file name.
1 parent daa9f54 commit 8228167

File tree

1 file changed

+13
-3
lines changed

1 file changed

+13
-3
lines changed

cmake/External/aotriton.cmake

Lines changed: 13 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -22,22 +22,31 @@ if(NOT __AOTRITON_INCLUDED)
2222
# Replaces .ci/docker/aotriton_version.txt
2323
# Note packages information may have versions skipped (due to no ABI breaks)
2424
# But they must be listed from lower version to higher version
25-
set(__AOTRITON_VER "0.9.2b")
25+
set(__AOTRITON_RELEASE_PAGE "0.9.2b")
26+
set(__AOTRITON_VER_LIST
27+
"0.9.2b" # rocm6.2
28+
"0.9.2b" # rocm6.3
29+
"0.9.2b" # rocm6.4
30+
"0.9.2b_612896439f" # rocm6.5 with gfx950
31+
)
2632
set(__AOTRITON_MANYLINUX_LIST
2733
"manylinux_2_28" # rocm6.2
2834
"manylinux_2_28" # rocm6.3
2935
"manylinux_2_28" # rocm6.4
36+
"manylinux_2_28" # rocm6.5
3037
)
3138
set(__AOTRITON_ROCM_LIST
3239
"rocm6.2"
3340
"rocm6.3"
3441
"rocm6.4"
42+
"rocm6.5"
3543
)
36-
set(__AOTRITON_CI_COMMIT "b388d223d8c7213545603e00f6f3148c54d1f525")
44+
set(__AOTRITON_CI_COMMIT "612896439fb4f78509b1a566b5ef0a333e9585bb") # source of rocm6.5 with gfx950
3745
set(__AOTRITON_SHA256_LIST
3846
"08d84f96f4c984179f80f517c0431c7511ee26bb0ce9bd05a827573ddd78cc79" # rocm6.2
3947
"9094d59717e7e6eace9126ca100dd0e86510f07fc6c3a349569fc4e2d9056604" # rocm6.3
4048
"41190202c2736d5ff75b13a3abc0fb52ebfbb67226cf85dc3de7699c7000db44" # rocm6.4
49+
"c85da64d21510190277794455ef8bd3f2d543a6f2462140d3da27e1df0ab8f82" # rocm6.5 with gfx950
4150
)
4251
set(__AOTRITON_Z "gz")
4352

@@ -93,13 +102,14 @@ if(NOT __AOTRITON_INCLUDED)
93102
list(FIND __AOTRITON_ROCM_LIST "rocm${__AOTRITON_ROCM}" __AOTRITON_ROCM_INDEX)
94103
list(GET __AOTRITON_SHA256_LIST ${__AOTRITON_ROCM_INDEX} __AOTRITON_SHA256)
95104
list(GET __AOTRITON_MANYLINUX_LIST ${__AOTRITON_ROCM_INDEX} __AOTRITON_MANYLINUX)
105+
list(GET __AOTRITON_VER_LIST ${__AOTRITON_ROCM_INDEX} __AOTRITON_VER)
96106
set(__AOTRITON_ARCH ${CMAKE_HOST_SYSTEM_PROCESSOR})
97107
string(CONCAT __AOTRITON_FILE "aotriton-"
98108
"${__AOTRITON_VER}-${__AOTRITON_MANYLINUX}"
99109
"_${__AOTRITON_ARCH}-rocm${__AOTRITON_ROCM}"
100110
"-shared.tar.${__AOTRITON_Z}")
101111
string(CONCAT __AOTRITON_URL "https://github.com/ROCm/aotriton/releases/download/"
102-
"${__AOTRITON_VER}/${__AOTRITON_FILE}")
112+
"${__AOTRITON_RELEASE_PAGE}/${__AOTRITON_FILE}")
103113
ExternalProject_Add(aotriton_external
104114
URL "${__AOTRITON_URL}"
105115
URL_HASH SHA256=${__AOTRITON_SHA256}

0 commit comments

Comments
 (0)