forked from pytorch/pytorch
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathbuild.bzl
104 lines (96 loc) · 2.55 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
93
94
95
96
97
98
99
100
101
102
103
104
def define_targets(rules):
rules.test_suite(
name = "tests",
tests = [
":core_tests",
":typeid_test",
":util_base_tests",
],
visibility = ["//:__pkg__"],
)
rules.cc_test(
name = "core_tests",
size = "small",
srcs = rules.glob([
"core/*.cpp",
"core/impl/*.cpp",
]),
copts = ["-Wno-deprecated-declarations"],
deps = [
"//c10/core:base",
"//c10/util:base",
"//c10/core:CPUAllocator",
"@com_google_googletest//:gtest_main",
],
)
rules.cc_test(
name = "typeid_test",
size = "small",
srcs = ["util/typeid_test.cpp"],
copts = ["-Wno-deprecated-declarations"],
deps = [
"//c10/util:typeid",
"@com_google_googletest//:gtest_main",
],
)
rules.cc_test(
name = "util_base_tests",
srcs = rules.glob(
["util/*.cpp"],
exclude = [
"util/bit_cast_test.cpp",
"util/ssize_test.cpp",
"util/typeid_test.cpp",
],
),
copts = ["-Wno-deprecated-declarations"],
deps = [
":Macros",
":complex_math_test_common",
":complex_test_common",
"//c10/macros",
"//c10/util:base",
"@com_google_googletest//:gtest_main",
],
)
rules.cc_test(
name = "util/bit_cast_test",
srcs = ["util/bit_cast_test.cpp"],
deps = [
"//c10/util:bit_cast",
"@com_google_googletest//:gtest_main",
],
)
rules.cc_test(
name = "util/ssize_test",
srcs = ["util/ssize_test.cpp"],
deps = [
"//c10/util:ssize",
"@com_google_googletest//:gtest_main",
],
)
rules.cc_library(
name = "Macros",
testonly = True,
hdrs = ["util/Macros.h"],
visibility = ["//:__subpackages__"],
)
rules.cc_library(
name = "complex_math_test_common",
testonly = True,
hdrs = ["util/complex_math_test_common.h"],
deps = [
"//c10/util:base",
"@com_google_googletest//:gtest",
],
)
rules.cc_library(
name = "complex_test_common",
testonly = True,
hdrs = ["util/complex_test_common.h"],
deps = [
"//c10/macros",
"//c10/util:base",
"@com_google_googletest//:gtest",
],
)