add tsl/protobuf:dnn_proto_cc_impl to xla_cc_binary

ddunl · tensorflower-gardener · commit fd9499348fb5 · 2022-12-15T11:20:00.000-08:00
PiperOrigin-RevId: 495641475
diff --git a/tensorflow/compiler/xla/hlo/experimental/auto_sharding/BUILD b/tensorflow/compiler/xla/hlo/experimental/auto_sharding/BUILD
@@ -147,6 +147,5 @@ xla_cc_binary(
         "//tensorflow/compiler/xla/stream_executor:dnn",
         "//tensorflow/compiler/xla/tools:hlo_module_loader",
         "//tensorflow/tsl/platform:platform_port",
-        "//tensorflow/tsl/protobuf:dnn_proto_cc_impl",
     ],
 )
diff --git a/tensorflow/compiler/xla/mlir/backends/cpu/BUILD b/tensorflow/compiler/xla/mlir/backends/cpu/BUILD
@@ -20,7 +20,6 @@ xla_cc_binary(
         "//tensorflow/compiler/xla/mlir_hlo:lhlo",
         "//tensorflow/compiler/xla/mlir_hlo:thlo",
         "//tensorflow/compiler/xla/service/cpu:cpu_compiler",
-        "//tensorflow/tsl/protobuf:dnn_proto_cc_impl",
         "@llvm-project//mlir:BufferizationTransforms",
         "@llvm-project//mlir:LinalgDialect",
         "@llvm-project//mlir:MlirOptLib",
diff --git a/tensorflow/compiler/xla/mlir/backends/gpu/BUILD b/tensorflow/compiler/xla/mlir/backends/gpu/BUILD
@@ -14,7 +14,6 @@ xla_cc_binary(
         "//tensorflow/compiler/xla/mlir_hlo:lhlo",
         "//tensorflow/compiler/xla/mlir_hlo:lhlo_gpu",
         "//tensorflow/compiler/xla/stream_executor:stream_executor_impl",
-        "//tensorflow/tsl/protobuf:dnn_proto_cc_impl",
         "@llvm-project//mlir:FuncDialect",
         "@llvm-project//mlir:GPUDialect",
         "@llvm-project//mlir:MemRefDialect",
diff --git a/tensorflow/compiler/xla/mlir/runtime/BUILD b/tensorflow/compiler/xla/mlir/runtime/BUILD
@@ -37,7 +37,6 @@ xla_cc_binary(
         "//tensorflow/compiler/xla/mlir/runtime/transforms:compilation_pipeline_cpu",
         "//tensorflow/compiler/xla/mlir/runtime/transforms:compilation_pipeline_gpu",
         "//tensorflow/compiler/xla/mlir/runtime/transforms:passes",
-        "//tensorflow/tsl/protobuf:dnn_proto_cc_impl",
         "@llvm-project//mlir:FuncDialect",
         "@llvm-project//mlir:MathDialect",
         "@llvm-project//mlir:MemRefDialect",
diff --git a/tensorflow/compiler/xla/rpc/BUILD b/tensorflow/compiler/xla/rpc/BUILD
@@ -70,7 +70,6 @@ xla_cc_binary(
     deps = [
         ":grpc_service_main_library",
         "//tensorflow/compiler/xla/service:cpu_plugin",
-        "//tensorflow/tsl/protobuf:dnn_proto_cc_impl",
     ],
 )
 
diff --git a/tensorflow/compiler/xla/service/gpu/tests/BUILD b/tensorflow/compiler/xla/service/gpu/tests/BUILD
@@ -827,7 +827,6 @@ xla_cc_binary(
         "//tensorflow/tsl/platform:platform_port",
         "//tensorflow/tsl/platform:tensor_float_32_utils",
         "//tensorflow/tsl/protobuf:autotuning_proto_cc_impl",
-        "//tensorflow/tsl/protobuf:dnn_proto_cc_impl",
         "//tensorflow/tsl/profiler/backends/cpu:annotation_stack_impl",
         "//tensorflow/tsl/util:command_line_flags",
         "//tensorflow/tsl/util:determinism",
diff --git a/tensorflow/compiler/xla/tests/BUILD b/tensorflow/compiler/xla/tests/BUILD
@@ -230,7 +230,6 @@ xla_cc_binary(
         "//tensorflow/compiler/xla/service/llvm_ir:llvm_util",
         "//tensorflow/tsl/platform:logging",
         "//tensorflow/tsl/platform:platform_port",
-        "//tensorflow/tsl/protobuf:dnn_proto_cc_impl",
         "@llvm-project//llvm:Support",
     ],
 )
diff --git a/tensorflow/compiler/xla/tools/BUILD b/tensorflow/compiler/xla/tools/BUILD
@@ -168,7 +168,6 @@ xla_cc_binary(
         "//tensorflow/tsl/platform:logging",
         "//tensorflow/tsl/platform:platform_port",
         "//tensorflow/tsl/platform:protobuf",
-        "//tensorflow/tsl/protobuf:dnn_proto_cc_impl",
     ],
 )
 
@@ -278,7 +277,6 @@ xla_cc_binary(
         "//tensorflow/tsl/platform:path",
         "//tensorflow/tsl/platform:platform_port",
         "//tensorflow/tsl/platform:subprocess",
-        "//tensorflow/tsl/protobuf:dnn_proto_cc_impl",
         "//tensorflow/tsl/protobuf:error_codes_proto_impl_cc",
         "//tensorflow/tsl/util:command_line_flags",
     ] + if_cuda_or_rocm([
diff --git a/tensorflow/compiler/xla/translate/BUILD b/tensorflow/compiler/xla/translate/BUILD
@@ -19,7 +19,6 @@ xla_cc_binary(
         "//tensorflow/compiler/xla/translate/mhlo_to_hlo:translate_registration",
         "//tensorflow/compiler/xla/translate/mhlo_to_lhlo_with_xla:translate_registration",
         "//tensorflow/tsl/platform:platform_port",
-        "//tensorflow/tsl/protobuf:dnn_proto_cc_impl",
         "@llvm-project//llvm:Support",
         "@llvm-project//mlir:IR",
         "@llvm-project//mlir:Support",
diff --git a/tensorflow/compiler/xla/translate/mhlo_to_lhlo_with_xla/BUILD b/tensorflow/compiler/xla/translate/mhlo_to_lhlo_with_xla/BUILD
@@ -77,7 +77,6 @@ xla_cc_binary(
         "//tensorflow/compiler/xla/service:cpu_plugin",
         "//tensorflow/compiler/xla/service/cpu:hlo_xla_runtime_pipeline",  # buildcleaner: keep
         "//tensorflow/tsl/platform:platform_port",
-        "//tensorflow/tsl/protobuf:dnn_proto_cc_impl",
         "@llvm-project//llvm:Support",
         "@llvm-project//mlir:AllPassesAndDialects",
         "@llvm-project//mlir:MlirOptLib",
@@ -98,7 +97,6 @@ xla_cc_binary(
         "//tensorflow/compiler/xla/service:gpu_plugin",
         "//tensorflow/compiler/xla/stream_executor:stream_executor_impl",
         "//tensorflow/tsl/platform:platform_port",
-        "//tensorflow/tsl/protobuf:dnn_proto_cc_impl",
         "@llvm-project//llvm:Support",
         "@llvm-project//mlir:AllPassesAndDialects",
         "@llvm-project//mlir:MlirOptLib",
diff --git a/tensorflow/compiler/xla/xla.bzl b/tensorflow/compiler/xla/xla.bzl
@@ -4,6 +4,7 @@ load(
     "//tensorflow/tsl:tsl.bzl",
     "clean_dep",
     "if_tsl_link_protobuf",
+    "tsl_copts",
 )
 load(
     "//tensorflow/tsl/platform/default:cuda_build_defs.bzl",
@@ -33,7 +34,7 @@ ORC_JIT_MEMORY_MAPPER_TARGETS = []
 def xla_py_test_deps():
     return []
 
-def xla_cc_binary(deps = None, **kwargs):
+def xla_cc_binary(deps = None, copts = tsl_copts(), **kwargs):
     if not deps:
         deps = []
 
@@ -50,8 +51,9 @@ def xla_cc_binary(deps = None, **kwargs):
         "//tensorflow/tsl/profiler/utils:time_utils_impl",
         "//tensorflow/tsl/profiler/backends/cpu:traceme_recorder_impl",
         "//tensorflow/tsl/protobuf:protos_all_cc_impl",
+        "//tensorflow/tsl/protobuf:dnn_proto_cc_impl",
     ]
-    native.cc_binary(deps = deps, **kwargs)
+    native.cc_binary(deps = deps, copts = copts, **kwargs)
 
 def xla_cc_test(
         name,

Original file line number	Diff line number	Diff line change
`@@ -147,6 +147,5 @@ xla_cc_binary(`
`147`	`147`	`"//tensorflow/compiler/xla/stream_executor:dnn",`
`148`	`148`	`"//tensorflow/compiler/xla/tools:hlo_module_loader",`
`149`	`149`	`"//tensorflow/tsl/platform:platform_port",`
`150`		`- "//tensorflow/tsl/protobuf:dnn_proto_cc_impl",`
`151`	`150`	`],`
`152`	`151`	`)`
Original file line number	Diff line number	Diff line change
`@@ -70,7 +70,6 @@ xla_cc_binary(`
`70`	`70`	`deps = [`
`71`	`71`	`":grpc_service_main_library",`
`72`	`72`	`"//tensorflow/compiler/xla/service:cpu_plugin",`
`73`		`- "//tensorflow/tsl/protobuf:dnn_proto_cc_impl",`
`74`	`73`	`],`
`75`	`74`	`)`
`76`	`75`
Original file line number	Diff line number	Diff line change
`@@ -230,7 +230,6 @@ xla_cc_binary(`
`230`	`230`	`"//tensorflow/compiler/xla/service/llvm_ir:llvm_util",`
`231`	`231`	`"//tensorflow/tsl/platform:logging",`
`232`	`232`	`"//tensorflow/tsl/platform:platform_port",`
`233`		`- "//tensorflow/tsl/protobuf:dnn_proto_cc_impl",`
`234`	`233`	`"@llvm-project//llvm:Support",`
`235`	`234`	`],`
`236`	`235`	`)`