@@ -28,14 +28,15 @@ set(XPU_XPTI_LIB_NAME "libxpti.so")
28
28
set (XPU_XBLAS_LIB_NAME "libxpu_blas.so" )
29
29
set (XPU_XFA_LIB_NAME "libxpu_flash_attention.so" )
30
30
set (XPU_XPUDNN_LIB_NAME "libxpu_dnn.so" )
31
+ set (XPU_XPUDNN_OMP_LIB_NAME "libomp.so" )
31
32
set (XPU_FFT_LIB_NAME "libcufft.so" )
32
33
# Avoid deprecated int32 apis:
33
34
add_compile_definitions (XPUAPI_NOT_INCLUDE_DEPRECATED)
34
35
35
36
if (NOT DEFINED XPU_XHPC_BASE_DATE)
36
- set (XPU_XHPC_BASE_DATE "dev/20990602 " )
37
+ set (XPU_XHPC_BASE_DATE "dev/20250722 " )
37
38
endif ()
38
- set (XPU_XCCL_BASE_VERSION "3.0.2.7 " ) # For XRE5
39
+ set (XPU_XCCL_BASE_VERSION "3.0.3.1 " ) # For XRE5
39
40
if (NOT DEFINED XPU_XFT_BASE_VERSION)
40
41
set (XPU_XFT_BASE_VERSION "20250507/xpu3" )
41
42
endif ()
@@ -183,6 +184,7 @@ set(XPU_CUDA_RT_LIB "${XPU_LIB_DIR}/${XPU_CUDA_RT_LIB_NAME}")
183
184
set (XPU_ML_LIB "${XPU_LIB_DIR} /${XPU_ML_LIB_NAME} " )
184
185
set (XPU_XFA_LIB "${XPU_LIB_DIR} /${XPU_XFA_LIB_NAME} " )
185
186
set (XPU_XPUDNN_LIB "${XPU_LIB_DIR} /${XPU_XPUDNN_LIB_NAME} " )
187
+ set (XPU_XPUDNN_OMP_LIB "${XPU_LIB_DIR} /${XPU_XPUDNN_OMP_LIB_NAME} " )
186
188
187
189
set (CMAKE_INSTALL_RPATH "${CMAKE_INSTALL_RPATH} " "${XPU_INSTALL_DIR} /lib" )
188
190
@@ -251,8 +253,9 @@ if(WITH_XPU_XRE5)
251
253
DOWNLOAD_COMMAND
252
254
bash ${CMAKE_SOURCE_DIR} /tools/xpu/pack_paddle_dependence.sh
253
255
${XPU_XRE_URL} ${XPU_XRE_DIR_NAME} ${XPU_XHPC_URL} ${XPU_XHPC_DIR_NAME}
254
- ${XPU_XCCL_URL} ${XPU_XCCL_DIR_NAME} 1 && wget ${XPU_XFT_GET_DEPENCE_URL}
255
- && bash ${XFT_COMMAND} ${XPU_XFT_URL} ${XPU_XFT_DIR_NAME} && bash
256
+ ${XPU_XCCL_URL} ${XPU_XCCL_DIR_NAME} 1 ${WITH_MKL} "${CMAKE_BINARY_DIR} "
257
+ && wget ${XPU_XFT_GET_DEPENCE_URL} && bash ${XFT_COMMAND} ${XPU_XFT_URL}
258
+ ${XPU_XFT_DIR_NAME} && bash
256
259
${CMAKE_SOURCE_DIR} /tools/xpu/get_xpti_dependence.sh ${XPU_XPTI_URL}
257
260
${XPU_XPTI_DIR_NAME} && bash
258
261
${CMAKE_SOURCE_DIR} /tools/xpu/get_xpufft_dependence.sh ${XPU_FFT_URL}
@@ -263,6 +266,7 @@ if(WITH_XPU_XRE5)
263
266
BUILD_BYPRODUCTS ${XPU_API_LIB}
264
267
BUILD_BYPRODUCTS ${XPU_XBLAS_LIB}
265
268
BUILD_BYPRODUCTS ${XPU_XPUDNN_LIB}
269
+ BUILD_BYPRODUCTS ${XPU_XPUDNN_OMP_LIB}
266
270
BUILD_BYPRODUCTS ${XPU_XFA_LIB}
267
271
BUILD_BYPRODUCTS ${XPU_RT_LIB}
268
272
BUILD_BYPRODUCTS ${XPU_CUDA_RT_LIB}
@@ -360,6 +364,10 @@ if(WITH_XPU_XRE5)
360
364
${XPU_XFA_LIB}
361
365
${XPU_XPUDNN_LIB}
362
366
${XPU_ML_LIB} )
367
+
368
+ if (NOT WITH_MKL)
369
+ target_link_libraries (xpulib ${XPU_XPUDNN_OMP_LIB} )
370
+ endif ()
363
371
else ()
364
372
target_link_libraries (xpulib ${XPU_RT_LIB} ${XPU_API_LIB} )
365
373
endif ()
0 commit comments