Skip to content

Commit 5592dff

Browse files
committed
[XPU] update bkcl to round mode
1 parent 30c0291 commit 5592dff

File tree

1 file changed

+13
-8
lines changed

1 file changed

+13
-8
lines changed

cmake/external/xpu.cmake

Lines changed: 13 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -28,24 +28,30 @@ set(XPU_XBLAS_LIB_NAME "libxpu_blas.so")
2828
set(XPU_XFA_LIB_NAME "libxpu_flash_attention.so")
2929
set(XPU_XPUDNN_LIB_NAME "libxpu_dnn.so")
3030

31-
if(NOT DEFINED XPU_XRE_BASE_VERSION)
32-
set(XPU_XRE_BASE_VERSION "4.32.0.1")
33-
endif()
3431
if(NOT DEFINED XPU_XHPC_BASE_DATE)
35-
set(XPU_XHPC_BASE_DATE "dev/20241113")
32+
set(XPU_XHPC_BASE_DATE "dev/20241115")
3633
endif()
37-
set(XPU_XCCL_BASE_VERSION "1.2.11e")
34+
set(XPU_XCCL_BASE_VERSION "3.0.0.5") # For XRE5
3835
if(NOT DEFINED XPU_XFT_BASE_VERSION)
3936
set(XPU_XFT_BASE_VERSION "20230602")
4037
endif()
38+
39+
if(NOT DEFINED XPU_XRE_BASE_VERSION)
40+
if(WITH_XPU_XRE5)
41+
set(XPU_XRE_BASE_VERSION "5.0.21.8")
42+
else()
43+
set(XPU_XRE_BASE_VERSION "4.32.0.1")
44+
endif()
45+
endif()
46+
4147
set(XPU_XPTI_BASE_VERSION "0.0.1")
4248

4349
set(XPU_XRE_BASE_URL
4450
"https://klx-sdk-release-public.su.bcebos.com/xre/release/${XPU_XRE_BASE_VERSION}"
4551
)
4652

4753
set(XPU_XCCL_BASE_URL
48-
"https://klx-sdk-release-public.su.bcebos.com/xccl/release/${XPU_XCCL_BASE_VERSION}.1"
54+
"https://klx-sdk-release-public.su.bcebos.com/xccl/release/2.0.0.1" # 2.0.0.1 is the final version for R200
4955
)
5056

5157
if(NOT XPU_XFT_BASE_URL)
@@ -62,12 +68,11 @@ if(WITH_XPTI)
6268
endif()
6369

6470
if(WITH_XPU_XRE5)
65-
set(XPU_XRE_BASE_VERSION "5.0.15.1")
6671
set(XPU_XRE_BASE_URL
6772
"https://klx-sdk-release-public.su.bcebos.com/xre/kl3-release/${XPU_XRE_BASE_VERSION}"
6873
)
6974
set(XPU_XCCL_BASE_URL
70-
"https://klx-sdk-release-public.su.bcebos.com/xccl/release/${XPU_XCCL_BASE_VERSION}.2"
75+
"https://klx-sdk-release-public.su.bcebos.com/xccl/release/${XPU_XCCL_BASE_VERSION}"
7176
)
7277
endif()
7378

0 commit comments

Comments
 (0)