mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 12:54:11 +08:00
e3f8141c25
Type punning using unions is undefined behavior in C++ (you may not access a member of a union that is not the active member). bit_cast is the right way. Pull Request resolved: https://github.com/pytorch/pytorch/pull/157942 Approved by: https://github.com/Skylion007
117 lines
2.9 KiB
Python
117 lines
2.9 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",
|
|
"//torch/headeronly:torch_headeronly",
|
|
"//c10/macros",
|
|
"@fmt",
|
|
"@moodycamel//:moodycamel",
|
|
] + 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__"],
|
|
deps = [
|
|
"//c10/macros",
|
|
],
|
|
)
|
|
|
|
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",
|
|
],
|
|
),
|
|
deps = [
|
|
"//torch/headeronly:torch_headeronly",
|
|
],
|
|
visibility = ["//visibility:public"],
|
|
)
|
|
|
|
rules.filegroup(
|
|
name = "headers",
|
|
srcs = rules.glob(
|
|
["*.h"],
|
|
exclude = [
|
|
"bit_cast.h",
|
|
"ssize.h",
|
|
],
|
|
),
|
|
visibility = [
|
|
"//:__pkg__",
|
|
"//c10:__pkg__",
|
|
],
|
|
)
|