experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

[OpenCL] Fixes run_metadata_test for SYCL

Browse Source 
This test is designed to test CUDA specific behavior
This commit is contained in:
Luke Iwanski 2017-06-22 19:31:25 +01:00
parent af0cbace1d
commit ad0892df1d
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/contrib/tfprof/python/tools/tfprof/internal/run_metadata_test.py
View File

@ -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'):