Fix a bug in the configure script that disables all GPU tests when GPU build is enabled.

PiperOrigin-RevId: 238155623
This commit is contained in:
Guangda Lai 2019-03-12 21:04:51 -07:00 committed by TensorFlower Gardener
parent f3954bf900
commit bcd701a0d4

View File

@ -1506,7 +1506,7 @@ def system_specific_test_config(env):
'test --test_tag_filters=-benchmark-test,-no_oss,-oss_serial') 'test --test_tag_filters=-benchmark-test,-no_oss,-oss_serial')
write_to_bazelrc('test --build_tag_filters=-benchmark-test,-no_oss') write_to_bazelrc('test --build_tag_filters=-benchmark-test,-no_oss')
if is_windows(): if is_windows():
if env.get('TF_NEED_CUDA', None) == 1: if env.get('TF_NEED_CUDA', None) == '1':
write_to_bazelrc( write_to_bazelrc(
'test --test_tag_filters=-no_windows,-no_windows_gpu,-no_gpu') 'test --test_tag_filters=-no_windows,-no_windows_gpu,-no_gpu')
write_to_bazelrc( write_to_bazelrc(
@ -1518,7 +1518,7 @@ def system_specific_test_config(env):
write_to_bazelrc('test --test_tag_filters=-gpu,-nomac,-no_mac') write_to_bazelrc('test --test_tag_filters=-gpu,-nomac,-no_mac')
write_to_bazelrc('test --build_tag_filters=-gpu,-nomac,-no_mac') write_to_bazelrc('test --build_tag_filters=-gpu,-nomac,-no_mac')
elif is_linux(): 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 --test_tag_filters=-no_gpu')
write_to_bazelrc('test --build_tag_filters=-no_gpu') write_to_bazelrc('test --build_tag_filters=-no_gpu')
write_to_bazelrc('test --test_env=LD_LIBRARY_PATH') write_to_bazelrc('test --test_env=LD_LIBRARY_PATH')