Ensure device_lib_test only runs GPU code when a GPU is available.

Change: 145342535
This commit is contained in:
Suharsh Sivakumar 2017-01-23 15:23:46 -08:00 committed by TensorFlower Gardener
parent 30d5f6f558
commit 6b2e92ed83

View File

@ -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])