Ensure device_lib_test only runs GPU code when a GPU is available.
Change: 145342535
This commit is contained in:
parent
30d5f6f558
commit
6b2e92ed83
@ -32,7 +32,7 @@ class DeviceLibTest(test_util.TensorFlowTestCase):
|
||||
self.assertEqual(devices[0].device_type, "CPU")
|
||||
|
||||
# GPU test
|
||||
if test.is_built_with_cuda():
|
||||
if test.is_gpu_available():
|
||||
self.assertGreater(len(devices), 1)
|
||||
self.assertTrue("GPU" in [d.device_type for d in devices])
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user