diff --git a/python/setup.py.in b/python/setup.py.in
index 669266d7b8b0603f30d369b12aad8bcc381e114e..6a252a5723d0daa232c304d315ecd3452a0b2de0 100644
--- a/python/setup.py.in
+++ b/python/setup.py.in
@@ -464,6 +464,10 @@ if '${WITH_LITE}' == 'ON':
             shutil.copy('${LITE_NNADAPTER_NPU_LIB}', libs_path)
             package_data['paddle.libs']+=['libnnadapter_driver_huawei_ascend_npu' + ext_name]
 
+if '${WITH_CINN}' == 'ON':
+    shutil.copy('${CINN_LIB_LOCATION}/${CINN_LIB_NAME}', libs_path)
+    package_data['paddle.libs']+=['libcinnapi.so']
+
 if '${WITH_PSLIB}' == 'ON':
     shutil.copy('${PSLIB_LIB}', libs_path)
     if os.path.exists('${PSLIB_VERSION_PY}'):