xref: /aosp_15_r20/external/pytorch/c10/build.bzl (revision da0073e96a02ea20f0ac840b70461e3646d07c45) 
1def define_targets(rules):
2    rules.cc_library(
3        name = "c10",
4        visibility = ["//visibility:public"],
5        deps = [
6            "//c10/core:CPUAllocator",
7            "//c10/core:ScalarType",
8            "//c10/core:alignment",
9            "//c10/core:alloc_cpu",
10            "//c10/core:base",
11            "//c10/macros",
12            "//c10/mobile:CPUCachingAllocator",
13            "//c10/mobile:CPUProfilingAllocator",
14            "//c10/util:TypeCast",
15            "//c10/util:base",
16            "//c10/util:typeid",
17        ] + rules.if_cuda(
18            [
19                "//c10/cuda:cuda",
20                "//c10/cuda:Macros",
21            ],
22            [],
23        ),
24    )
25