forked from pytorch/pytorch
-
Notifications
You must be signed in to change notification settings - Fork 5
Expand file tree
/
Copy pathbuild.bzl
More file actions
24 lines (24 loc) · 706 Bytes
/
build.bzl
File metadata and controls
24 lines (24 loc) · 706 Bytes
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
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"],
)