From 643ff03fbc5e100176070760c9c73f11a695caf3 Mon Sep 17 00:00:00 2001
From: Yancey <yanxu05@baidu.com>
Date: Fri, 5 Jan 2018 18:51:49 +0800
Subject: [PATCH] capi package (#7237)

---
 paddle/scripts/docker/build.sh | 30 +++++++++++++++++++++++++++---
 1 file changed, 27 insertions(+), 3 deletions(-)

diff --git a/paddle/scripts/docker/build.sh b/paddle/scripts/docker/build.sh
index 92039ec6b05..f1e244772f3 100644
--- a/paddle/scripts/docker/build.sh
+++ b/paddle/scripts/docker/build.sh
@@ -193,6 +193,25 @@ EOF
 EOF
 }
 
+function gen_capi_package() {
+  if [[ ${WITH_C_API} == "ON" ]]; then
+    install_prefix="/paddle/build/capi_output"
+    rm -rf $install_prefix
+
+    make DESTDIR="$install_prefix" install
+
+    if [[ ${WITH_MKL:-OFF} == "ON" ]]; then
+      find ./third_party/install -name 'libmklml_gnu.so' -exec cp {} $install_prefix/usr/local/lib \;
+      find ./third_party/install -name 'libmklml_intel.so' -exec cp {} $install_prefix/usr/local/lib \;
+      cp -P ./third_party/install/mkldnn/lib/* $install_prefix/usr/local/lib/
+    fi
+
+    find ./third_party/install -name 'libiomp5.so' -exec cp {} $install_prefix/usr/local/lib \;
+    cd $install_prefix/usr/local
+    ls | egrep -v "^Found.*item$" | xargs tar /paddle/build/paddle.tgz
+  fi
+}
+
 set -xe
 
 cmake_gen ${PYTHON_ABI:-""}
@@ -200,6 +219,11 @@ run_build
 run_test
 gen_docs
 gen_dockerfile
-
-printf "If you need to install PaddlePaddle in develop docker image,"
-printf "please make install or pip install build/python/dist/*.whl.\n"
+gen_capi_package
+
+if [[ ${WITH_C_API:-OFF} == "ON" ]]; then
+  printf "PaddlePaddle C-API libraries was generated on build/paddle.tgz\n" 
+else
+  printf "If you need to install PaddlePaddle in develop docker image,"
+  printf "please make install or pip install build/python/dist/*.whl.\n"
+fi
-- 
GitLab