From 5d8284c38dd6f30b1091c6d1decf11ce59d55df0 Mon Sep 17 00:00:00 2001
From: jiangcheng <thisjiang@qq.com>
Date: Fri, 23 Dec 2022 10:44:13 +0800
Subject: [PATCH] CINN copy float16.h to package libs dir (#49191)

---
 setup.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/setup.py b/setup.py
index 6abd61c002..55853ad9ed 100644
--- a/setup.py
+++ b/setup.py
@@ -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
-- 
GitLab