From f4be9d6a32043cdfccd0103e8925ce29c2171d20 Mon Sep 17 00:00:00 2001 From: QingshuChen Date: Mon, 28 Dec 2020 17:57:29 +0800 Subject: [PATCH] add bkcl.so in whl for kunlun (#29947) --- python/setup.py.in | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/python/setup.py.in b/python/setup.py.in index c732a892167..b29d91caf7c 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}' -- GitLab