add WITH_DOC for print_operators_doc in docker/build.sh

8 years ago · 3e8c3638dc
parent 45dfc34c21
commit 3e8c3638dc
1 changed files with 8 additions and 1 deletions
--- a/paddle/scripts/docker/build.sh
+++ b/paddle/scripts/docker/build.sh
@ -185,7 +185,14 @@ EOF
    ${DOCKERFILE_GPU_ENV}
    ADD go/cmd/pserver/pserver /usr/bin/
    ADD go/cmd/master/master /usr/bin/
-    ADD paddle/pybind/print_operators_doc /usr/bin/
+EOF
+
+    if [[ ${WITH_DOC:-OFF} == 'ON' ]]; then
+        cat >> /paddle/build/Dockerfile <<EOF
+        ADD paddle/pybind/print_operators_doc /usr/bin/
+EOF
+    fi
+    cat >> /paddle/build/Dockerfile <<EOF
    # default command shows the paddle version and exit
    CMD ["paddle", "version"]
 EOF