Fix kokoro build script redirection (#4521)
diff --git a/build_tools/kokoro/gcp_ubuntu/cmake-bazel/linux/x86-turing/build.sh b/build_tools/kokoro/gcp_ubuntu/cmake-bazel/linux/x86-turing/build.sh index bf199b4..5d1e909 100755 --- a/build_tools/kokoro/gcp_ubuntu/cmake-bazel/linux/x86-turing/build.sh +++ b/build_tools/kokoro/gcp_ubuntu/cmake-bazel/linux/x86-turing/build.sh
@@ -36,9 +36,6 @@ ./build_tools/kokoro/gcp_ubuntu/check_vulkan.sh -# Print SwiftShader git commit -cat /swiftshader/git-commit - # TODO(4195): TensorFlow integrations can only be built out-of-tree (and kokoro # does not support that). # Temporarily get around these conflicting restrictions by nesting the git
diff --git a/build_tools/kokoro/gcp_ubuntu/cmake-bazel/linux/x86-turing/build_kokoro.sh b/build_tools/kokoro/gcp_ubuntu/cmake-bazel/linux/x86-turing/build_kokoro.sh index 53b443b..f7a5c89 100755 --- a/build_tools/kokoro/gcp_ubuntu/cmake-bazel/linux/x86-turing/build_kokoro.sh +++ b/build_tools/kokoro/gcp_ubuntu/cmake-bazel/linux/x86-turing/build_kokoro.sh
@@ -37,6 +37,7 @@ docker_setup docker run "${DOCKER_RUN_ARGS[@]?}" \ + --gpus all \ gcr.io/iree-oss/cmake-bazel-tensorflow-nvidia@sha256:1592a4ffe32ca5643b27f59f3fc4052d96a5bea34a91e51703d882aa38c09ae5 \ build_tools/kokoro/gcp_ubuntu/cmake-bazel/linux/x86-turing/build.sh
diff --git a/build_tools/kokoro/gcp_ubuntu/cmake-bazel/linux/x86-turing/common.cfg b/build_tools/kokoro/gcp_ubuntu/cmake-bazel/linux/x86-turing/common.cfg index 187b7fc..365606f 100644 --- a/build_tools/kokoro/gcp_ubuntu/cmake-bazel/linux/x86-turing/common.cfg +++ b/build_tools/kokoro/gcp_ubuntu/cmake-bazel/linux/x86-turing/common.cfg
@@ -14,4 +14,4 @@ # See the License for the specific language governing permissions and # limitations under the License. -build_file: "iree/build_tools/kokoro/gcp_ubuntu/cmake-bazel/linux/x86-swiftshader/build_kokoro.sh" +build_file: "iree/build_tools/kokoro/gcp_ubuntu/cmake-bazel/linux/x86-turing/build_kokoro.sh"