pytorch

1
def 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