aboutsummaryrefslogtreecommitdiffhomepage
diff options
context:
space:
mode:
-rwxr-xr-xconfigure9
1 files changed, 7 insertions, 2 deletions
diff --git a/configure b/configure
index 1531b4d40a..4bcf11e0a9 100755
--- a/configure
+++ b/configure
@@ -714,8 +714,13 @@ if is_windows; then
write_to_bazelrc "test --config=win-cuda"
else
# If CUDA is enabled, always use GPU during build and test.
- write_to_bazelrc "build --config=cuda"
- write_to_bazelrc "test --config=cuda"
+ if [ "$TF_CUDA_CLANG" == "1" ]; then
+ write_to_bazelrc "build --config=cuda_clang"
+ write_to_bazelrc "test --config=cuda_clang"
+ else
+ write_to_bazelrc "build --config=cuda"
+ write_to_bazelrc "test --config=cuda"
+ fi
fi
# end of if "$TF_NEED_CUDA" == "1"