diff options
author | 2018-06-18 13:08:29 -0500 | |
---|---|---|
committer | 2018-06-18 13:08:29 -0500 | |
commit | 9ac856f65798d008da2fc2ca6c9041748474ccfe (patch) | |
tree | 4f0a0036408357751d4497a1693559c9784dbe31 /tensorflow/tools/ci_build/ci_build.sh | |
parent | ff7e6399443615675a3f1182c4f2e1850008da04 (diff) |
cpu and gpu Dockerfiles for ppc64le
Adding Dockerfile.cpu.ppc64le and Dockerfile.gpu.ppc64le to enable the ability
to do builds using docker on ppc64le. Also enables the ability to run
ci_sanity.sh (from ci_build.sh) on ppc64le.
Modified ci_build.sh and ci_parameterized_build.sh to accept container types
that start with cpu or gpu.
Added install_bazel_from_source.sh and install_buildifier_from_source.sh install
scripts to avoid installing x86 versions of the binaries. These scripts could be
used by other platforms in the future.
Diffstat (limited to 'tensorflow/tools/ci_build/ci_build.sh')
-rwxr-xr-x | tensorflow/tools/ci_build/ci_build.sh | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/tools/ci_build/ci_build.sh b/tensorflow/tools/ci_build/ci_build.sh index 1f0fd0387a..f6a50d3d4c 100755 --- a/tensorflow/tools/ci_build/ci_build.sh +++ b/tensorflow/tools/ci_build/ci_build.sh @@ -79,7 +79,7 @@ if [[ "${CONTAINER_TYPE}" == "cmake" ]]; then fi # Use nvidia-docker if the container is GPU. -if [[ "${CONTAINER_TYPE}" == "gpu" ]]; then +if [[ "${CONTAINER_TYPE}" == gpu* ]]; then DOCKER_BINARY="nvidia-docker" else DOCKER_BINARY="docker" @@ -99,7 +99,7 @@ BUILD_TAG="${BUILD_TAG:-tf_ci}" # Add extra params for cuda devices and libraries for GPU container. # And clear them if we are not building for GPU. -if [[ "${CONTAINER_TYPE}" != "gpu" ]]; then +if [[ "${CONTAINER_TYPE}" != gpu* ]]; then GPU_EXTRA_PARAMS="" fi |