mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
acaf6ba3c6
Summary: Undo highlevel BUCKification in favor of something more organized by moving it to the dir itself Test Plan: CI Rollback Plan: Reviewed By: swolchok Differential Revision: D76920013 Pull Request resolved: https://github.com/pytorch/pytorch/pull/156503 Approved by: https://github.com/swolchok
122 lines
3.1 KiB
Python
122 lines
3.1 KiB
Python
def define_targets(rules):
|
|
rules.cc_library(
|
|
name = "CPUAllocator",
|
|
srcs = ["CPUAllocator.cpp"],
|
|
hdrs = ["CPUAllocator.h"],
|
|
linkstatic = True,
|
|
local_defines = ["C10_BUILD_MAIN_LIB"],
|
|
visibility = ["//visibility:public"],
|
|
deps = [
|
|
":alignment",
|
|
":base",
|
|
"//c10/mobile:CPUCachingAllocator",
|
|
"//c10/mobile:CPUProfilingAllocator",
|
|
"//c10/util:base",
|
|
],
|
|
# This library defines a flag, The use of alwayslink keeps it
|
|
# from being stripped.
|
|
alwayslink = True,
|
|
)
|
|
|
|
rules.cc_library(
|
|
name = "ScalarType",
|
|
hdrs = ["ScalarType.h"],
|
|
linkstatic = True,
|
|
local_defines = ["C10_BUILD_MAIN_LIB"],
|
|
visibility = ["//visibility:public"],
|
|
deps = ["//c10/util:base"],
|
|
)
|
|
|
|
rules.cc_library(
|
|
name = "alignment",
|
|
hdrs = ["alignment.h"],
|
|
linkstatic = True,
|
|
local_defines = ["C10_BUILD_MAIN_LIB"],
|
|
visibility = ["//visibility:public"],
|
|
)
|
|
|
|
rules.cc_library(
|
|
name = "alloc_cpu",
|
|
srcs = ["impl/alloc_cpu.cpp"],
|
|
hdrs = ["impl/alloc_cpu.h"],
|
|
linkstatic = True,
|
|
local_defines = ["C10_BUILD_MAIN_LIB"],
|
|
visibility = ["//visibility:public"],
|
|
deps = [
|
|
":alignment",
|
|
"//c10/macros",
|
|
"//c10/util:base",
|
|
],
|
|
# This library defines flags, The use of alwayslink keeps them
|
|
# from being stripped.
|
|
alwayslink = True,
|
|
)
|
|
|
|
rules.cc_library(
|
|
name = "base",
|
|
srcs = rules.glob(
|
|
[
|
|
"*.cpp",
|
|
"impl/*.cpp",
|
|
],
|
|
exclude = [
|
|
"CPUAllocator.cpp",
|
|
"impl/alloc_cpu.cpp",
|
|
],
|
|
),
|
|
hdrs = rules.glob(
|
|
[
|
|
"*.h",
|
|
"impl/*.h",
|
|
],
|
|
exclude = [
|
|
"CPUAllocator.h",
|
|
"impl/alloc_cpu.h",
|
|
],
|
|
),
|
|
linkstatic = True,
|
|
local_defines = ["C10_BUILD_MAIN_LIB"],
|
|
visibility = ["//visibility:public"],
|
|
deps = [
|
|
":ScalarType",
|
|
"//third_party/cpuinfo",
|
|
"//c10/macros",
|
|
"//c10/util:TypeCast",
|
|
"//c10/util:base",
|
|
"//c10/util:typeid",
|
|
],
|
|
# This library uses flags and registration. Do not let the
|
|
# linker remove them.
|
|
alwayslink = True,
|
|
)
|
|
|
|
rules.cc_library(
|
|
name = "base_headers",
|
|
srcs = [],
|
|
hdrs = rules.glob(
|
|
[
|
|
"*.h",
|
|
"impl/*.h",
|
|
],
|
|
exclude = [
|
|
"CPUAllocator.h",
|
|
"impl/alloc_cpu.h",
|
|
],
|
|
),
|
|
visibility = ["//visibility:public"],
|
|
)
|
|
|
|
rules.filegroup(
|
|
name = "headers",
|
|
srcs = rules.glob(
|
|
[
|
|
"*.h",
|
|
"impl/*.h",
|
|
],
|
|
exclude = [
|
|
"alignment.h",
|
|
],
|
|
),
|
|
visibility = ["//visibility:public"],
|
|
)
|