diff options
Diffstat (limited to 'tensorflow/core/common_runtime/direct_session_test.cc')
-rw-r--r-- | tensorflow/core/common_runtime/direct_session_test.cc | 3 |
1 files changed, 1 insertions, 2 deletions
diff --git a/tensorflow/core/common_runtime/direct_session_test.cc b/tensorflow/core/common_runtime/direct_session_test.cc index f8deaaf222..3d06ca0ae4 100644 --- a/tensorflow/core/common_runtime/direct_session_test.cc +++ b/tensorflow/core/common_runtime/direct_session_test.cc @@ -877,8 +877,6 @@ class BlockingOp : public OpKernel { REGISTER_KERNEL_BUILDER(Name("BlockingOp").Device(DEVICE_CPU), BlockingOp); REGISTER_OP("BlockingOp").Input("x: float").Output("y: float").Doc(""); -REGISTER_KERNEL_BUILDER(Name("BlockingOp").Device(DEVICE_SYCL), BlockingOp); - static void TestSessionInterOpThreadsImpl(bool use_function_lib) { FunctionDefLibrary library_graph_def; if (use_function_lib) { @@ -916,6 +914,7 @@ static void TestSessionInterOpThreadsImpl(bool use_function_lib) { ->set_opt_level(OptimizerOptions_Level_L0); (*options.config.mutable_device_count())["CPU"] = 2; (*options.config.mutable_device_count())["GPU"] = 0; + (*options.config.mutable_device_count())["SYCL"] = 0; options.config.add_session_inter_op_thread_pool(); auto* p = options.config.add_session_inter_op_thread_pool(); |