mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 12:54:11 +08:00
66f53889d5
Summary: nativert RFC: https://github.com/zhxchen17/rfcs/blob/master/RFC-0043-torch-native-runtime.md To land the runtime into PyTorch core, we will gradually land logical parts of the code into the Github issue and get each piece properly reviewed. This diff adds a simple semaphore interface into c10 until c++20 where we get counting_semaphore gonna need a oss build export to take a look at this... Test Plan: CI Differential Revision: D73882656 Pull Request resolved: https://github.com/pytorch/pytorch/pull/153504 Approved by: https://github.com/zhxchen17
54 lines
1.4 KiB
Plaintext
54 lines
1.4 KiB
Plaintext
load("//tools/build_defs:glob_defs.bzl", "subdir_glob")
|
|
|
|
cxx_library(
|
|
name = "c10",
|
|
srcs = glob(
|
|
["**/*.cpp"],
|
|
exclude = [
|
|
"test/**/*.cpp",
|
|
"benchmark/**/*.cpp",
|
|
"cuda/**/*.cpp",
|
|
"xpu/**/*.cpp",
|
|
],
|
|
),
|
|
deps = [
|
|
"//third_party:cpuinfo",
|
|
"//third_party:fmt",
|
|
"//third_party:glog",
|
|
"//third_party:moodycamel",
|
|
],
|
|
exported_deps = [],
|
|
compiler_flags = [
|
|
"-Werror",
|
|
"-Wno-global-constructors",
|
|
"-DDISABLE_NAMEDTENSOR",
|
|
"-DSUPPORTS_BACKTRACE=0"
|
|
],
|
|
exported_headers = subdir_glob(
|
|
[
|
|
("", "**/*.h"),
|
|
],
|
|
exclude = [
|
|
"test/**/*.h",
|
|
"benchmark/**/*.h",
|
|
"cuda/**/*.h",
|
|
"xpu/**/*.h",
|
|
],
|
|
),
|
|
exported_linker_flags = [],
|
|
exported_preprocessor_flags = [
|
|
'-DC10_USING_CUSTOM_GENERATED_MACROS',
|
|
'-DC10_USE_GLOG',
|
|
'-DC10_USE_MINIMAL_GLOG',
|
|
'-DC10_MOBILE',
|
|
'-fexceptions',
|
|
'-Wno-global-constructors'
|
|
],
|
|
header_namespace = "c10",
|
|
link_whole = True,
|
|
platform_preprocessor_flags = [['windows', ['-D_WINDOWS', '-D_WIN32', '-DWIN32', '-DNOMINMAX', '-D_CRT_SECURE_NO_WARNINGS', '-D_USE_MATH_DEFINES']], ['windows.*64$', ['-D_WIN64']]],
|
|
preprocessor_flags = ['-DC10_BUILD_MAIN_LIB'],
|
|
reexport_all_header_dependencies = True,
|
|
visibility = ['PUBLIC'],
|
|
)
|