mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 12:54:11 +08:00
cce4d213a6
It depends on /c10/util:base which is not header-only. Differential Revision: [D76552750](https://our.internmc.facebook.com/intern/diff/D76552750/) **NOTE FOR REVIEWERS**: This PR has internal Meta-specific changes or comments, please review them on [Phabricator](https://our.internmc.facebook.com/intern/diff/D76552750/)! Pull Request resolved: https://github.com/pytorch/pytorch/pull/155858 Approved by: https://github.com/ezyang
36 lines
973 B
Python
36 lines
973 B
Python
def define_targets(rules):
|
|
rules.cc_library(
|
|
name = "c10",
|
|
visibility = ["//visibility:public"],
|
|
deps = [
|
|
"//c10/core:CPUAllocator",
|
|
"//c10/core:ScalarType",
|
|
"//c10/core:alignment",
|
|
"//c10/core:alloc_cpu",
|
|
"//c10/core:base",
|
|
"//c10/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",
|
|
],
|
|
[],
|
|
),
|
|
)
|
|
|
|
rules.cc_library(
|
|
name = "c10_headers",
|
|
deps = [
|
|
"//c10/core:base_headers",
|
|
"//c10/macros",
|
|
"//c10/util:base_headers",
|
|
"//c10/util:bit_cast",
|
|
],
|
|
visibility = ["//visibility:public"],
|
|
)
|