1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
|
"""Wrapper around cc_proto_library used inside the XLA codebase."""
load("@protobuf_archive//:protobuf.bzl", "cc_proto_library")
# xla_proto_library() is a convenience wrapper around cc_proto_library.
def xla_proto_library(name, srcs=[], deps=[], visibility=None, testonly=0):
cc_proto_library(name=name,
srcs=srcs,
deps=deps,
cc_libs = ["@protobuf_archive//:protobuf"],
protoc="@protobuf_archive//:protoc",
default_runtime="@protobuf_archive//:protobuf",
testonly=testonly,
visibility=visibility,)
# Flags required for modules that export symbols that are to be called by the
# XLA CustomCall operator. CustomCall must be able to find symbols with dlsym(),
# which on Linux requires we link with --export-dynamic.
export_dynamic_linkopts = select({
"//tensorflow:darwin": [],
"//conditions:default": ["-Wl,--export-dynamic"],
})
|
