xref: /aosp_15_r20/external/pytorch/c10/test/build.bzl (revision da0073e96a02ea20f0ac840b70461e3646d07c45) 
1def define_targets(rules):
2    rules.test_suite(
3        name = "tests",
4        tests = [
5            ":core_tests",
6            ":typeid_test",
7            ":util_base_tests",
8        ],
9        visibility = ["//:__pkg__"],
10    )
11
12    rules.cc_test(
13        name = "core_tests",
14        size = "small",
15        srcs = rules.glob([
16            "core/*.cpp",
17            "core/impl/*.cpp",
18        ]),
19        copts = ["-Wno-deprecated-declarations"],
20        deps = [
21            "//c10/core:base",
22            "//c10/util:base",
23            "//c10/core:CPUAllocator",
24            "@com_google_googletest//:gtest_main",
25        ],
26    )
27
28    rules.cc_test(
29        name = "typeid_test",
30        size = "small",
31        srcs = ["util/typeid_test.cpp"],
32        copts = ["-Wno-deprecated-declarations"],
33        deps = [
34            "//c10/util:typeid",
35            "@com_google_googletest//:gtest_main",
36        ],
37    )
38
39    rules.cc_test(
40        name = "util_base_tests",
41        srcs = rules.glob(
42            ["util/*.cpp"],
43            exclude = [
44                "util/bit_cast_test.cpp",
45                "util/ssize_test.cpp",
46                "util/typeid_test.cpp",
47            ],
48        ),
49        copts = ["-Wno-deprecated-declarations"],
50        deps = [
51            ":Macros",
52            ":complex_math_test_common",
53            ":complex_test_common",
54            "//c10/macros",
55            "//c10/util:base",
56            "@com_google_googletest//:gtest_main",
57        ],
58    )
59
60    rules.cc_test(
61        name = "util/bit_cast_test",
62        srcs = ["util/bit_cast_test.cpp"],
63        deps = [
64            "//c10/util:bit_cast",
65            "@com_google_googletest//:gtest_main",
66        ],
67    )
68
69    rules.cc_test(
70        name = "util/ssize_test",
71        srcs = ["util/ssize_test.cpp"],
72        deps = [
73            "//c10/util:ssize",
74            "@com_google_googletest//:gtest_main",
75        ],
76    )
77
78    rules.cc_library(
79        name = "Macros",
80        testonly = True,
81        hdrs = ["util/Macros.h"],
82        visibility = ["//:__subpackages__"],
83    )
84
85    rules.cc_library(
86        name = "complex_math_test_common",
87        testonly = True,
88        hdrs = ["util/complex_math_test_common.h"],
89        deps = [
90            "//c10/util:base",
91            "@com_google_googletest//:gtest",
92        ],
93    )
94
95    rules.cc_library(
96        name = "complex_test_common",
97        testonly = True,
98        hdrs = ["util/complex_test_common.h"],
99        deps = [
100            "//c10/macros",
101            "//c10/util:base",
102            "@com_google_googletest//:gtest",
103        ],
104    )
105