add bkcl.so in whl for kunlun (#29947)

f4be9d6a · QingshuChen · GitHub · 726c78f2 · f4be9d6a
隐藏空白更改
内联并排

Showing with 4 addition and 0 deletion

python/setup.py.in python/setup.py.in +4 -0

未找到文件。
--- 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}'