1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
|
# This file is licensed under the Apache License v2.0 with LLVM Exceptions.
# See https://llvm.org/LICENSE.txt for license information.
# SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
"""Rules and macros for MLIR"""
def if_cuda_available(if_true, if_false = []):
return select({
# CUDA is not yet supported.
"//conditions:default": if_false,
})
def _cc_headers_only_impl(ctx):
return CcInfo(compilation_context = ctx.attr.src[CcInfo].compilation_context)
cc_headers_only = rule(
implementation = _cc_headers_only_impl,
attrs = {
"src": attr.label(
mandatory = True,
providers = [CcInfo],
),
},
doc = "Provides the headers from 'src' without linking anything.",
provides = [CcInfo],
)
def mlir_c_api_cc_library(
name,
srcs = [],
hdrs = [],
deps = [],
header_deps = [],
capi_deps = [],
**kwargs):
"""Macro that generates three targets for MLIR C API libraries.
* A standard cc_library target ("Name"),
* A header-only cc_library target ("NameHeaders")
* An implementation cc_library target tagged `alwayslink` suitable for
inclusion in a shared library built with cc_binary() ("NameObjects").
In order to avoid duplicate symbols, it is important that
mlir_c_api_cc_library targets only depend on other mlir_c_api_cc_library
targets via the "capi_deps" parameter. This makes it so that "FooObjects"
depend on "BarObjects" targets and "Foo" targets depend on "Bar" targets.
Don't cross the streams.
"""
capi_header_deps = ["%sHeaders" % d for d in capi_deps]
capi_object_deps = ["%sObjects" % d for d in capi_deps]
native.cc_library(
name = name,
srcs = srcs,
hdrs = hdrs,
deps = deps + capi_deps + header_deps,
**kwargs
)
native.cc_library(
name = name + "Headers",
hdrs = hdrs,
deps = header_deps + capi_header_deps,
**kwargs
)
native.cc_library(
name = name + "Objects",
srcs = srcs,
hdrs = hdrs,
deps = deps + capi_object_deps + capi_header_deps + header_deps,
alwayslink = True,
**kwargs
)
|
