diff options
Diffstat (limited to 'tensorflow/core/ops/math_grad_test.cc')
| -rw-r--r-- | tensorflow/core/ops/math_grad_test.cc | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/tensorflow/core/ops/math_grad_test.cc b/tensorflow/core/ops/math_grad_test.cc index b3b523001b..1fc5a1b128 100644 --- a/tensorflow/core/ops/math_grad_test.cc +++ b/tensorflow/core/ops/math_grad_test.cc @@ -388,6 +388,9 @@ class TestOp : public OpKernel { void Compute(OpKernelContext* ctx) override { ctx->set_output(0, Tensor()); } }; REGISTER_KERNEL_BUILDER(Name("TestOpWithNoGrad").Device(DEVICE_CPU), TestOp); +#ifdef TENSORFLOW_USE_SYCL +REGISTER_KERNEL_BUILDER(Name("TestOpWithNoGrad").Device(DEVICE_SYCL), TestOp); +#endif // TENSORFLOW_USE_SYCL TEST_F(MathGradTest, Error_Reporting) { auto x = test::AsTensor<float>({-3.f}); |