Skip to content

Commit 643ff03

Browse files
author
Yancey
authored
capi package (#7237)
1 parent 673dc26 commit 643ff03

File tree

1 file changed

+27
-3
lines changed

1 file changed

+27
-3
lines changed

paddle/scripts/docker/build.sh

Lines changed: 27 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -193,13 +193,37 @@ EOF
193193
EOF
194194
}
195195

196+
function gen_capi_package() {
197+
if [[ ${WITH_C_API} == "ON" ]]; then
198+
install_prefix="/paddle/build/capi_output"
199+
rm -rf $install_prefix
200+
201+
make DESTDIR="$install_prefix" install
202+
203+
if [[ ${WITH_MKL:-OFF} == "ON" ]]; then
204+
find ./third_party/install -name 'libmklml_gnu.so' -exec cp {} $install_prefix/usr/local/lib \;
205+
find ./third_party/install -name 'libmklml_intel.so' -exec cp {} $install_prefix/usr/local/lib \;
206+
cp -P ./third_party/install/mkldnn/lib/* $install_prefix/usr/local/lib/
207+
fi
208+
209+
find ./third_party/install -name 'libiomp5.so' -exec cp {} $install_prefix/usr/local/lib \;
210+
cd $install_prefix/usr/local
211+
ls | egrep -v "^Found.*item$" | xargs tar /paddle/build/paddle.tgz
212+
fi
213+
}
214+
196215
set -xe
197216

198217
cmake_gen ${PYTHON_ABI:-""}
199218
run_build
200219
run_test
201220
gen_docs
202221
gen_dockerfile
203-
204-
printf "If you need to install PaddlePaddle in develop docker image,"
205-
printf "please make install or pip install build/python/dist/*.whl.\n"
222+
gen_capi_package
223+
224+
if [[ ${WITH_C_API:-OFF} == "ON" ]]; then
225+
printf "PaddlePaddle C-API libraries was generated on build/paddle.tgz\n"
226+
else
227+
printf "If you need to install PaddlePaddle in develop docker image,"
228+
printf "please make install or pip install build/python/dist/*.whl.\n"
229+
fi

0 commit comments

Comments
 (0)