fix

Author: dynamicheart

cmake/external/xpu.cmake

@@ -22,6 +22,7 @@ set(XPU_API_LIB_NAME "libxpuapi.so")
 set(XPU_RT_LIB_NAME "libxpurt.so")
 set(XPU_CUDA_LIB_NAME "libxpucuda.so")
 set(XPU_CUDA_RT_LIB_NAME "libcudart.so")
+set(XPU_ML_LIB_NAME "libxpuml.so")
 set(XPU_XFT_LIB_NAME "libxft.so")
 set(XPU_XPTI_LIB_NAME "libxpti.so")
 set(XPU_XBLAS_LIB_NAME "libxpu_blas.so")
@@ -146,6 +147,7 @@ set(XPU_XBLAS_LIB "${XPU_LIB_DIR}/${XPU_XBLAS_LIB_NAME}")
 set(XPU_RT_LIB "${XPU_LIB_DIR}/${XPU_RT_LIB_NAME}")
 set(XPU_CUDA_LIB "${XPU_LIB_DIR}/${XPU_CUDA_LIB_NAME}")
 set(XPU_CUDA_RT_LIB "${XPU_LIB_DIR}/${XPU_CUDA_RT_LIB_NAME}")
+set(XPU_ML_LIB "${XPU_LIB_DIR}/${XPU_ML_LIB_NAME}")
 set(XPU_XFA_LIB "${XPU_LIB_DIR}/${XPU_XFA_LIB_NAME}")
 set(XPU_XPUDNN_LIB "${XPU_LIB_DIR}/${XPU_XPUDNN_LIB_NAME}")
 
@@ -190,6 +192,7 @@ if(WITH_XPU_XRE5)
     BUILD_BYPRODUCTS ${XPU_XFA_LIB}
     BUILD_BYPRODUCTS ${XPU_RT_LIB}
     BUILD_BYPRODUCTS ${XPU_CUDA_RT_LIB}
+    BUILD_BYPRODUCTS ${XPU_ML_LIB}
     BUILD_BYPRODUCTS ${XPU_BKCL_LIB})
 else()
   ExternalProject_Add(

python/env_dict.py.in

@@ -65,6 +65,8 @@ env_dict={
     'WITH_XPU_XRE5':'@WITH_XPU_XRE5@',
     'XPU_CUDA_RT_LIB':'@XPU_CUDA_RT_LIB@',
     'XPU_CUDA_RT_LIB_NAME':'@XPU_CUDA_RT_LIB_NAME@',
+    'XPU_ML_LIB':'@XPU_ML_LIB@',
+    'XPU_ML_LIB_NAME':'@XPU_ML_LIB_NAME@',
     'WITH_XPU_BKCL':'@WITH_XPU_BKCL@',
     'XPU_BKCL_LIB':'@XPU_BKCL_LIB@',
     'XPU_BKCL_LIB_NAME':'@XPU_BKCL_LIB_NAME@',

python/setup.py.in

@@ -1027,6 +1027,10 @@ if '${WITH_XPU}' == 'ON':
         for xpu_cuda_rt_lib_file in xpu_cuda_rt_lib_list:
             shutil.copy(xpu_cuda_rt_lib_file, libs_path)
             package_data['paddle.libs'] += [os.path.basename(xpu_cuda_rt_lib_file)]
+        xpu_ml_lib_list = glob.glob('${XPU_ML_LIB}*')
+        for xpu_ml_lib_file in xpu_ml_lib_list:
+            shutil.copy(xpu_ml_lib_file, libs_path)
+            package_data['paddle.libs'] += [os.path.basename(xpu_ml_lib_file)]
         shutil.copy('${XPU_XBLAS_LIB}', libs_path)
         package_data['paddle.libs'] += ['${XPU_XBLAS_LIB_NAME}']
         shutil.copy('${XPU_XFA_LIB}', libs_path)

setup.py

@@ -1482,6 +1482,12 @@ def get_package_data_and_package_dir():
                 package_data['paddle.libs'] += [
                     os.path.basename(xpu_cuda_rt_lib_file)
                 ]
+            xpu_ml_lib_list = glob.glob(env_dict.get("XPU_ML_LIB") + '*')
+            for xpu_ml_lib_file in xpu_ml_lib_list:
+                shutil.copy(xpu_ml_lib_file, libs_path)
+                package_data['paddle.libs'] += [
+                    os.path.basename(xpu_ml_lib_file)
+                ]
             shutil.copy(env_dict.get("XPU_XBLAS_LIB"), libs_path)
             package_data['paddle.libs'] += [env_dict.get("XPU_XBLAS_LIB_NAME")]
             shutil.copy(env_dict.get("XPU_XFA_LIB"), libs_path)

tools/xpu/pack_paddle_dependence.sh

@@ -64,6 +64,7 @@ function xre_prepare() {
   check_files ${XRE_DIR_NAME}/include/xpu/runtime.h ${XRE_DIR_NAME}/so/libxpurt.so
   if [ "$WITH_XPU_XRE5" -eq 1 ]; then
     check_files ${XRE_DIR_NAME}/so/libcudart.so
+    check_files ${XRE_DIR_NAME}/so/libxpuml.so
   fi
   cp -r ${XRE_DIR_NAME}/include/xpu/* xpu/include/xpu/
   cp -r ${XRE_DIR_NAME}/so/* xpu/lib/