When GPU is configured, selection of config to set depends on value of TF_CUDA_CLANG (--config=cuda / --config=cuda_clang)

PiperOrigin-RevId: 160433337

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"