Skip to content

Commit ad0892d

Browse files
author
Luke Iwanski
committed
[OpenCL] Fixes run_metadata_test for SYCL
This test is designed to test CUDA specific behavior
1 parent af0cbac commit ad0892d

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tensorflow/contrib/tfprof/python/tools/tfprof/internal/run_metadata_test.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -71,7 +71,7 @@ def _run_model():
7171
class RunMetadataTest(test.TestCase):
7272

7373
def testGPU(self):
74-
if not test.is_gpu_available():
74+
if not test.is_gpu_available(cuda_only=True):
7575
return
7676

7777
with ops.device('/gpu:0'):

0 commit comments

Comments
 (0)