mirror of
https://github.com/saymrwulf/pytorch.git
synced 2026-05-14 20:57:59 +00:00
ade8fee512
Summary: X-link: https://github.com/pytorch/executorch/pull/7040 Accomplished by importing relevant files from c10 into executorch/runtime/core/portable_type/c10, and then using `using` in the top-level ExecuTorch headers. This approach should keep the ExecuTorch build hermetic for embedded use cases. In the future, we should add a CI job to ensure the c10 files stay identical to the PyTorch ones. ghstack-source-id: 260047850 exported-using-ghexport Test Plan: builds Differential Revision: D66106969 Pull Request resolved: https://github.com/pytorch/pytorch/pull/144111 Approved by: https://github.com/malfet
108 lines
2.7 KiB
Python
108 lines
2.7 KiB
Python
def define_targets(rules):
|
|
rules.cc_library(
|
|
name = "TypeCast",
|
|
srcs = ["TypeCast.cpp"],
|
|
hdrs = ["TypeCast.h"],
|
|
linkstatic = True,
|
|
local_defines = ["C10_BUILD_MAIN_LIB"],
|
|
visibility = ["//visibility:public"],
|
|
deps = [
|
|
":base",
|
|
"//c10/core:ScalarType",
|
|
"//c10/macros",
|
|
],
|
|
)
|
|
|
|
rules.cc_library(
|
|
name = "base",
|
|
srcs = rules.glob(
|
|
["*.cpp"],
|
|
exclude = [
|
|
"TypeCast.cpp",
|
|
"typeid.cpp",
|
|
],
|
|
),
|
|
hdrs = rules.glob(
|
|
["*.h"],
|
|
exclude = [
|
|
"TypeCast.h",
|
|
"typeid.h",
|
|
],
|
|
),
|
|
linkstatic = True,
|
|
local_defines = ["C10_BUILD_MAIN_LIB"],
|
|
visibility = ["//visibility:public"],
|
|
deps = [
|
|
":bit_cast",
|
|
"//c10/macros",
|
|
"@fmt",
|
|
] + rules.select({
|
|
"//c10:using_gflags": ["@com_github_gflags_gflags//:gflags"],
|
|
"//conditions:default": [],
|
|
}) + rules.select({
|
|
"//c10:using_glog": ["@com_github_glog//:glog"],
|
|
"//conditions:default": [],
|
|
}),
|
|
linkopts = rules.select({
|
|
"@bazel_tools//src/conditions:windows": [],
|
|
"//conditions:default": ["-ldl"],
|
|
}),
|
|
# This library uses flags and registration. Do not let the
|
|
# linker remove them.
|
|
alwayslink = True,
|
|
)
|
|
|
|
rules.cc_library(
|
|
name = "bit_cast",
|
|
hdrs = ["bit_cast.h"],
|
|
visibility = ["//:__subpackages__"],
|
|
)
|
|
|
|
rules.cc_library(
|
|
name = "ssize",
|
|
hdrs = ["ssize.h"],
|
|
linkstatic = True,
|
|
visibility = ["//:__subpackages__"],
|
|
deps = [":base"],
|
|
)
|
|
|
|
rules.cc_library(
|
|
name = "typeid",
|
|
srcs = ["typeid.cpp"],
|
|
hdrs = ["typeid.h"],
|
|
linkstatic = True,
|
|
local_defines = ["C10_BUILD_MAIN_LIB"],
|
|
visibility = ["//visibility:public"],
|
|
deps = [
|
|
":base",
|
|
"//c10/core:ScalarType",
|
|
"//c10/macros",
|
|
],
|
|
)
|
|
|
|
rules.cc_library(
|
|
name = "base_headers",
|
|
hdrs = rules.glob(
|
|
["*.h"],
|
|
exclude = [
|
|
"bit_cast.h",
|
|
"ssize.h",
|
|
],
|
|
),
|
|
visibility = ["//visibility:public"],
|
|
)
|
|
|
|
rules.filegroup(
|
|
name = "headers",
|
|
srcs = rules.glob(
|
|
["*.h"],
|
|
exclude = [
|
|
"bit_cast.h",
|
|
"ssize.h",
|
|
],
|
|
),
|
|
visibility = [
|
|
"//:__pkg__",
|
|
"//c10:__pkg__",
|
|
],
|
|
)
|