forked from pytorch/pytorch
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathbuild.bzl
92 lines (87 loc) · 2.27 KB
/
build.bzl
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
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
def define_targets(rules):
rules.cc_library(
name = "TypeCast",
srcs = ["TypeCast.cpp"],
hdrs = ["TypeCast.h"],
linkstatic = True,
local_defines = ["C10_BUILD_MAIN_LIB"],
visibility = ["//visibility:public"],
deps = [
":base",
"//c10/core:ScalarType",
"//c10/macros",
],
)
rules.cc_library(
name = "base",
srcs = rules.glob(
["*.cpp"],
exclude = [
"TypeCast.cpp",
"typeid.cpp",
],
),
hdrs = rules.glob(
["*.h"],
exclude = [
"TypeCast.h",
"typeid.h",
],
),
linkstatic = True,
local_defines = ["C10_BUILD_MAIN_LIB"],
visibility = ["//visibility:public"],
deps = [
":bit_cast",
"//c10/macros",
"@fmt",
] + rules.select({
"//c10:using_gflags": ["@com_github_gflags_gflags//:gflags"],
"//conditions:default": [],
}) + rules.select({
"//c10:using_glog": ["@com_github_glog//:glog"],
"//conditions:default": [],
}),
# This library uses flags and registration. Do not let the
# linker remove them.
alwayslink = True,
)
rules.cc_library(
name = "bit_cast",
hdrs = ["bit_cast.h"],
visibility = ["//:__subpackages__"],
)
rules.cc_library(
name = "ssize",
hdrs = ["ssize.h"],
linkstatic = True,
visibility = ["//:__subpackages__"],
deps = [":base"],
)
rules.cc_library(
name = "typeid",
srcs = ["typeid.cpp"],
hdrs = ["typeid.h"],
linkstatic = True,
local_defines = ["C10_BUILD_MAIN_LIB"],
visibility = ["//visibility:public"],
deps = [
":base",
"//c10/core:ScalarType",
"//c10/macros",
],
)
rules.filegroup(
name = "headers",
srcs = rules.glob(
["*.h"],
exclude = [
"bit_cast.h",
"ssize.h",
],
),
visibility = [
"//:__pkg__",
"//c10:__pkg__",
],
)