Files
pytorch/c10/build.bzl

Ignoring revisions in .git-blame-ignore-revs. Click here to bypass and see the normal blame view.

25 lines
706 B
Python
Raw Normal View History

def define_targets(rules):
rules.cc_library(
name = "c10",
deps = [
"//c10/core:CPUAllocator",
"//c10/core:ScalarType",
"//c10/core:alignment",
"//c10/core:alloc_cpu",
"//c10/core:base",
"//c10/macros:macros",
"//c10/mobile:CPUCachingAllocator",
"//c10/mobile:CPUProfilingAllocator",
"//c10/util:TypeCast",
"//c10/util:base",
"//c10/util:typeid",
] + rules.if_cuda(
[
"//c10/cuda:cuda",
"//c10/cuda:Macros",
],
[],
),
visibility = ["//visibility:public"],
)