@@ -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