Skip to content

Commit

Permalink
[OpenCL] Fixes CUDA specific test run on SYCL (#56)
Browse files Browse the repository at this point in the history
The testBadParentValuesOnGPU should only be run on CUDA devices, as the
test checks for particular CUDA behaviour. We don't actually provide a
SYCL kernel for GatherTree and so it's not a problem that the tests
don't target SYCL.
  • Loading branch information
jwlawson authored and Luke Iwanski committed Jun 22, 2017
1 parent af0cbac commit 1699d90
Showing 1 changed file with 3 additions and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -65,7 +65,9 @@ def testBadParentValuesOnCPU(self):
_ = beams.eval()

def testBadParentValuesOnGPU(self):
if not test.is_gpu_available():
# Only want to run this test on CUDA devices, as gather_tree is not
# registered for SYCL devices.
if not test.is_gpu_available(cuda_only=True):
return
# (max_time = 4, batch_size = 1, beams = 3)
# bad parent in beam 1 time 1; appears as a negative index at time 0
Expand Down

0 comments on commit 1699d90

Please sign in to comment.