diff --git a/python/setup.py.in b/python/setup.py.in index c732a8921670e6e9f8e66f0ce4a567603ab010c5..b29d91caf7ced7c39ce79b04a970bfefa53bc5eb 100644 --- a/python/setup.py.in +++ b/python/setup.py.in @@ -322,6 +322,10 @@ if '${WITH_XPU}' == 'ON': package_data['paddle.libs']+=['${XPU_API_LIB_NAME}', '${XPU_RT_LIB_NAME}'] +if '${WITH_XPU_BKCL}' == 'ON': + shutil.copy('${XPU_BKCL_LIB}', libs_path) + package_data['paddle.libs']+=['${XPU_BKCL_LIB_NAME}'] + # copy libfuild_framework.so to libs if os.name != 'nt' and sys.platform != 'darwin': paddle_framework_lib='${FLUID_FRAMEWORK_SHARED_LIB}'