add train demo to paddle_build.sh (#19885)
* add train demo to paddle_build.sh test=develop test=document_preview * code cleaning * test=develop test=develop * remove program desc files test=develop * add test_train to CI test=develop test=document_preview * make run.sh excutable test=develop test=document_preview * switch with_mkl to off test=develop test=document_preview * switch with_mkl to off test=develop * switch with_mkl to on test=develop * move test_fluid_lib_train to ci_py35 test=develop * fix bugs of MKLDNN building * rm build_train_lib(), test=develop * restore tensor.h, test=developfix-python-transpose
parent
e7a6567be5
commit
dd8fc9810d
@ -0,0 +1,4 @@
|
||||
set -x
|
||||
cd `dirname $0`
|
||||
rm -rf build/
|
||||
set +x
|
@ -0,0 +1,30 @@
|
||||
#!/bin/bash
|
||||
|
||||
set -x
|
||||
|
||||
PADDLE_ROOT=$1
|
||||
TURN_ON_MKL=$2 # use MKL or Openblas
|
||||
|
||||
# download models
|
||||
function download() {
|
||||
wget -q http://paddle-tar.bj.bcebos.com/train_demo/LR/main_program
|
||||
wget -q http://paddle-tar.bj.bcebos.com/train_demo/LR/startup_program
|
||||
}
|
||||
|
||||
download
|
||||
|
||||
# build demo trainer
|
||||
fluid_install_dir=${PADDLE_ROOT}/build/fluid_install_dir
|
||||
|
||||
mkdir -p build
|
||||
cd build
|
||||
rm -rf *
|
||||
cmake .. -DPADDLE_LIB=$fluid_install_dir \
|
||||
-DWITH_MKLDNN=$TURN_ON_MKL \
|
||||
-DWITH_MKL=$TURN_ON_MKL
|
||||
make
|
||||
|
||||
cd ..
|
||||
|
||||
# run demo trainer
|
||||
build/demo_trainer
|
Loading…
Reference in new issue