Enable CUDA on remainder of eligible bots. (#8868)
Adds to #8847
diff --git a/build_tools/kokoro/gcp_ubuntu/cmake-bazel/linux/x86-swiftshader/build.sh b/build_tools/kokoro/gcp_ubuntu/cmake-bazel/linux/x86-swiftshader/build.sh
index 702db3b..b6c9bb8 100755
--- a/build_tools/kokoro/gcp_ubuntu/cmake-bazel/linux/x86-swiftshader/build.sh
+++ b/build_tools/kokoro/gcp_ubuntu/cmake-bazel/linux/x86-swiftshader/build.sh
@@ -62,6 +62,8 @@
-DIREE_BUILD_TESTS=ON \
-DIREE_BUILD_SAMPLES=OFF \
-DIREE_BUILD_PYTHON_BINDINGS=ON \
+ -DIREE_HAL_DRIVER_CUDA=ON \
+ -DIREE_TARGET_BACKEND_CUDA=ON \
.
echo "Building with Ninja"
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 0baddcd..200be0f 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
@@ -57,8 +57,6 @@
# TODO(gcmn): It would be nice to be able to build and test as much as possible,
# so a build failure only prevents building/testing things that depend on it and
# we can still run the other tests.
-# TODO: Add "-DIREE_TARGET_BACKEND_CUDA=ON -DIREE_HAL_DRIVER_CUDA=ON" once the
-# VMs have been updated with the correct CUDA SDK.
echo "Configuring CMake"
"${CMAKE_BIN}" -B "${CMAKE_BUILD_DIR?}" -G Ninja \
-DIREE_TF_TOOLS_ROOT="${BAZEL_BINDIR?}/iree_tf_compiler/" \
@@ -67,6 +65,8 @@
-DIREE_BUILD_TESTS=ON \
-DIREE_BUILD_SAMPLES=OFF \
-DIREE_BUILD_PYTHON_BINDINGS=ON \
+ -DIREE_HAL_DRIVER_CUDA=ON \
+ -DIREE_TARGET_BACKEND_CUDA=ON \
.
echo "Building with Ninja"
diff --git a/build_tools/kokoro/gcp_ubuntu/cmake/linux/x86-swiftshader-asan/build.sh b/build_tools/kokoro/gcp_ubuntu/cmake/linux/x86-swiftshader-asan/build.sh
index 1864778..b6bdca6 100755
--- a/build_tools/kokoro/gcp_ubuntu/cmake/linux/x86-swiftshader-asan/build.sh
+++ b/build_tools/kokoro/gcp_ubuntu/cmake/linux/x86-swiftshader-asan/build.sh
@@ -36,6 +36,12 @@
"-DCMAKE_BUILD_TYPE=RelWithDebInfo"
"-DIREE_ENABLE_ASAN=ON"
"-B" "${CMAKE_BUILD_DIR?}"
+
+ # Enable CUDA compiler and runtime builds unconditionally. Our CI images all
+ # have enough deps to at least build CUDA support and compile CUDA binaries
+ # (but not necessarily test on real hardware).
+ "-DIREE_HAL_DRIVER_CUDA=ON"
+ "-DIREE_TARGET_BACKEND_CUDA=ON"
)
echo "Configuring CMake"