Skip to content

Commit 6ab41eb

Browse files
authored
[XPU] aarch64 support XCCL (#75797)
Update XPU_XCCL_BASE_VERSION for ARM configuration.
1 parent d887c61 commit 6ab41eb

File tree

1 file changed

+13
-4
lines changed

1 file changed

+13
-4
lines changed

cmake/external/xpu.cmake

Lines changed: 13 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -36,7 +36,11 @@ add_compile_definitions(XPUAPI_NOT_INCLUDE_DEPRECATED)
3636
if(NOT DEFINED XPU_XHPC_BASE_DATE)
3737
set(XPU_XHPC_BASE_DATE "dev/20251031")
3838
endif()
39-
set(XPU_XCCL_BASE_VERSION "3.0.3.4") # For XRE5
39+
if(WITH_ARM)
40+
set(XPU_XCCL_BASE_VERSION "20251104") # For XRE5
41+
else()
42+
set(XPU_XCCL_BASE_VERSION "3.0.3.4") # For XRE5
43+
endif()
4044
if(NOT DEFINED XPU_XFT_BASE_VERSION)
4145
set(XPU_XFT_BASE_VERSION "20250507/xpu3")
4246
endif()
@@ -56,7 +60,11 @@ else()
5660
endif()
5761

5862
if(NOT DEFINED XPU_FFT_BASE_DATE)
59-
set(XPU_FFT_BASE_DATE "20250704")
63+
if(WITH_ARM)
64+
set(XPU_FFT_BASE_DATE "20251017")
65+
else()
66+
set(XPU_FFT_BASE_DATE "20250704")
67+
endif()
6068
endif()
6169

6270
set(XPU_XRE_BASE_URL
@@ -97,7 +105,7 @@ if(WITH_ARM)
97105
if(WITH_XPU_XRE5)
98106
set(XPU_XRE_DIR_NAME "xre-kylin_v10_server-aarch64-${XPU_XRE_BASE_VERSION}")
99107
# TODO: xccl has no kylin output now. set default value here.
100-
set(XPU_XCCL_DIR_NAME "xccl_Linux_x86_64")
108+
set(XPU_XCCL_DIR_NAME "xccl_Linux_aarch64")
101109
set(XPU_XHPC_DIR_NAME "xhpc-kylinv4_aarch64")
102110
set(XPU_XFT_DIR_NAME "") # TODO: xft has no kylin output at now.
103111
else()
@@ -106,7 +114,8 @@ if(WITH_ARM)
106114
set(XPU_XFT_DIR_NAME "") # TODO: xft has no kylin output at now.
107115
endif()
108116
if(WITH_XPU_FFT)
109-
set(XPU_FFT_DIR_NAME "kylin_v10_aarch64/xpufft_kylinv10_aarch64_v2")
117+
set(XPU_FFT_DIR_NAME
118+
"kylin_v10_aarch64/${XPU_FFT_BASE_DATE}/xpufft_kylinv10_aarch64")
110119
endif()
111120
elseif(WITH_SUNWAY)
112121
set(XPU_XRE_DIR_NAME "xre-deepin_sw6_64")

0 commit comments

Comments
 (0)