未验证 提交 5d8284c3 编写于 作者: J jiangcheng 提交者: GitHub

CINN copy float16.h to package libs dir (#49191)

上级 df5587dc
......@@ -855,8 +855,13 @@ def get_package_data_and_package_dir():
+ '/cinn/runtime/cuda/cinn_cuda_runtime_source.cuh',
libs_path,
)
shutil.copy(
env_dict.get("CINN_INCLUDE_DIR") + '/cinn/runtime/cuda/float16.h',
libs_path,
)
package_data['paddle.libs'] += ['libcinnapi.so']
package_data['paddle.libs'] += ['cinn_cuda_runtime_source.cuh']
package_data['paddle.libs'] += ['float16.h']
if env_dict.get("CMAKE_BUILD_TYPE") == 'Release' and os.name != 'nt':
command = (
"patchelf --set-rpath '$ORIGIN/' %s/" % libs_path
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册