mirror of
https://github.com/zebrajr/tensorflow.git
synced 2025-12-06 12:20:11 +01:00
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:
parent
f3954bf900
commit
bcd701a0d4
|
|
@ -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')
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue
Block a user