[OpenCL] Fixes run_metadata_test for SYCL

This test is designed to test CUDA specific behavior
lukeiwanski · Jun 22, 2017 · ad0892d · ad0892d
1 parent af0cbac
commit ad0892d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tensorflow/contrib/tfprof/python/tools/tfprof/internal/run_metadata_test.py b/tensorflow/contrib/tfprof/python/tools/tfprof/internal/run_metadata_test.py
@@ -71,7 +71,7 @@ def _run_model():
 class RunMetadataTest(test.TestCase):
 
   def testGPU(self):
-    if not test.is_gpu_available():
+    if not test.is_gpu_available(cuda_only=True):
       return
 
     with ops.device('/gpu:0'):