Skip to content

Commit f019f51

Browse files
authored
Merge pull request tensorflow#10993 from lukeiwanski/opencl/run_metadata_test
[OpenCL] Fixes run_metadata_test for SYCL
2 parents f302975 + ad0892d commit f019f51

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)