Fix a bug in the configure script that disables all GPU tests when GPU build is enabled.
PiperOrigin-RevId: 238155623
diff --git a/configure.py b/configure.py
index 08684e8..3dc3969 100644
--- a/configure.py
+++ b/configure.py
@@ -1506,7 +1506,7 @@
'test --test_tag_filters=-benchmark-test,-no_oss,-oss_serial')
write_to_bazelrc('test --build_tag_filters=-benchmark-test,-no_oss')
if is_windows():
- if env.get('TF_NEED_CUDA', None) == 1:
+ if env.get('TF_NEED_CUDA', None) == '1':
write_to_bazelrc(
'test --test_tag_filters=-no_windows,-no_windows_gpu,-no_gpu')
write_to_bazelrc(
@@ -1518,7 +1518,7 @@
write_to_bazelrc('test --test_tag_filters=-gpu,-nomac,-no_mac')
write_to_bazelrc('test --build_tag_filters=-gpu,-nomac,-no_mac')
elif is_linux():
- if env.get('TF_NEED_CUDA', None) == 1:
+ if env.get('TF_NEED_CUDA', None) == '1':
write_to_bazelrc('test --test_tag_filters=-no_gpu')
write_to_bazelrc('test --build_tag_filters=-no_gpu')
write_to_bazelrc('test --test_env=LD_LIBRARY_PATH')