@@ -560,8 +560,6 @@ filegroup(
560560 "client/**/*.cc" ,
561561 "common_runtime/**/*.h" ,
562562 "common_runtime/**/*.cc" ,
563- "debug/**/*.h" ,
564- "debug/**/*.cc" ,
565563 "framework/**/*.h" ,
566564 "framework/**/*.cc" ,
567565 "graph/**/*.h" ,
@@ -1087,7 +1085,6 @@ tf_cuda_library(
10871085 linkstatic = 1 ,
10881086 deps = [
10891087 ":core_cpu_internal" ,
1090- ":debug_graph_utils" ,
10911088 ":framework" ,
10921089 ":gpu_tracer" ,
10931090 ":lib" ,
@@ -1117,23 +1114,6 @@ tf_cuda_library(
11171114 alwayslink = 1 ,
11181115)
11191116
1120- tf_cuda_library (
1121- name = "debug_graph_utils" ,
1122- srcs = ["debug/debug_graph_utils.cc" ],
1123- hdrs = ["debug/debug_graph_utils.h" ],
1124- copts = tf_copts (),
1125- linkstatic = 1 ,
1126- deps = [
1127- ":core_cpu_internal" ,
1128- ":framework" ,
1129- ":lib" ,
1130- ":lib_internal" ,
1131- ":proto_text" ,
1132- ":protos_all_cc" ,
1133- ],
1134- alwayslink = 1 ,
1135- )
1136-
11371117cc_library (
11381118 name = "example_parser_configuration" ,
11391119 srcs = ["example/example_parser_configuration.cc" ],
@@ -1603,9 +1583,43 @@ tf_cc_test(
16031583 ],
16041584)
16051585
1606- tf_cc_test_gpu (
1586+ tf_cc_test (
16071587 name = "debug/debug_gateway_test" ,
16081588 size = "small" ,
1589+ linkstatic = tf_kernel_tests_linkstatic (),
1590+ deps = [
1591+ ":core" ,
1592+ ":core_cpu" ,
1593+ ":core_cpu_internal" ,
1594+ ":debug_gateway_internal" ,
1595+ ":direct_session_internal" ,
1596+ ":framework" ,
1597+ ":framework_internal" ,
1598+ ":lib" ,
1599+ ":lib_internal" ,
1600+ ":ops" ,
1601+ ":protos_all_cc" ,
1602+ ":test" ,
1603+ ":test_main" ,
1604+ ":testlib" ,
1605+ "//tensorflow/cc:cc_ops" ,
1606+ "//tensorflow/core/kernels:control_flow_ops" ,
1607+ "//tensorflow/core/kernels:cwise_op" ,
1608+ "//tensorflow/core/kernels:dense_update_ops" ,
1609+ "//tensorflow/core/kernels:fifo_queue_op" ,
1610+ "//tensorflow/core/kernels:identity_op" ,
1611+ "//tensorflow/core/kernels:matmul_op" ,
1612+ "//tensorflow/core/kernels:ops_util" ,
1613+ "//tensorflow/core/kernels:queue_ops" ,
1614+ "//tensorflow/core/kernels:session_ops" ,
1615+ "//tensorflow/core/kernels:variable_ops" ,
1616+ "//third_party/eigen3" ,
1617+ ],
1618+ )
1619+
1620+ tf_cc_test_gpu (
1621+ name = "debug/debug_gateway_gpu_test" ,
1622+ size = "small" ,
16091623 args = ["--heap_check=local" ],
16101624 linkstatic = tf_kernel_tests_linkstatic (),
16111625 tags = tf_cuda_tests_tags () + ["nomac" ],
@@ -1614,7 +1628,6 @@ tf_cc_test_gpu(
16141628 ":core_cpu" ,
16151629 ":core_cpu_internal" ,
16161630 ":debug_gateway_internal" ,
1617- ":debug_graph_utils" ,
16181631 ":direct_session" ,
16191632 ":direct_session_internal" ,
16201633 ":framework" ,
@@ -1627,7 +1640,6 @@ tf_cc_test_gpu(
16271640 ":test_main" ,
16281641 ":testlib" ,
16291642 "//tensorflow/cc:cc_ops" ,
1630- "//tensorflow/core/kernels:debug_ops" ,
16311643 "//tensorflow/core/kernels:ops_util" ,
16321644 ],
16331645)
0 commit comments