diff options
| author |  Yifei Feng <yifeif@google.com> | 2018-05-15 10:58:43 -0700 |
|---|---|---|
| committer | Yifei Feng <yifeif@google.com> | 2018-05-15 10:58:43 -0700 |
| commit | 41ed2381d9d958abf6c93eb8e49e88282a5099ae (patch) | |
| tree | 03142a327497f65199f8c9ae86062de1ce1c1e71 /tensorflow/tools/pip_package/build_pip_package.sh | |
| parent | de4a6e646be56ca59c78dd6f92f8f6bcc7196696 (diff) | |
| parent | a7a3bb3df12c632b81bf1b23f8405f92a0c903c3 (diff) | |
Merge commit for internal changes
Diffstat (limited to 'tensorflow/tools/pip_package/build_pip_package.sh')
| -rwxr-xr-x | tensorflow/tools/pip_package/build_pip_package.sh | 11 |
1 files changed, 10 insertions, 1 deletions
diff --git a/tensorflow/tools/pip_package/build_pip_package.sh b/tensorflow/tools/pip_package/build_pip_package.sh index 3af79ee170..1a83c6e757 100755 --- a/tensorflow/tools/pip_package/build_pip_package.sh +++ b/tensorflow/tools/pip_package/build_pip_package.sh @@ -53,6 +53,7 @@ function main() { PKG_NAME_FLAG="" GPU_BUILD=0 NIGHTLY_BUILD=0 + PROJECT_NAME="" while true; do if [[ "$1" == "--nightly_flag" ]]; then NIGHTLY_BUILD=1 @@ -60,6 +61,12 @@ function main() { GPU_BUILD=1 elif [[ "$1" == "--gpudirect" ]]; then PKG_NAME_FLAG="--project_name tensorflow_gpudirect" + elif [[ "$1" == "--project_name" ]]; then + shift + if [[ -z "$1" ]]; then + break + fi + PROJECT_NAME="$1" fi shift @@ -68,7 +75,9 @@ function main() { fi done - if [[ ${NIGHTLY_BUILD} == "1" && ${GPU_BUILD} == "1" ]]; then + if [[ -n ${PROJECT_NAME} ]]; then + PKG_NAME_FLAG="--project_name ${PROJECT_NAME}" + elif [[ ${NIGHTLY_BUILD} == "1" && ${GPU_BUILD} == "1" ]]; then PKG_NAME_FLAG="--project_name tf_nightly_gpu" elif [[ ${NIGHTLY_BUILD} == "1" ]]; then PKG_NAME_FLAG="--project_name tf_nightly" |