"""Wrapper around cc_proto_library used inside the XLA codebase."""

load("//tensorflow/core:platform/default/build_config.bzl",
     "cc_proto_library")
load("//tensorflow/core:platform/default/build_config_root.bzl",
     "if_static")

# xla_proto_library() is a convenience wrapper around cc_proto_library.
def xla_proto_library(name, srcs=[], deps=[], visibility=None, testonly=0, **kwargs):
  if kwargs.get('use_grpc_plugin'):
    kwargs['use_grpc_namespace'] = True
  cc_proto_library(name=name,
                   srcs=srcs,
                   deps=deps,
                   cc_libs = if_static(
                       ["@protobuf_archive//:protobuf"],
                       otherwise=["@protobuf_archive//:protobuf_headers"],
                   ),
                   protoc="@protobuf_archive//:protoc",
                   testonly=testonly,
                   visibility=visibility,
                   **kwargs)

def xla_py_grpc_library(**kwargs):
  # Note: we don't currently define any special targets for Python GRPC in OSS.
  _ignore = kwargs
  pass


ORC_JIT_MEMORY_MAPPER_TARGETS = []