xref: /aosp_15_r20/external/pytorch/third_party/BUCK.oss (revision da0073e96a02ea20f0ac840b70461e3646d07c45)

load("//third_party:glog.buck.bzl", "define_glog")
load("//third_party:xnnpack.buck.bzl", "define_xnnpack")
load("//third_party:kineto.buck.bzl", "define_kineto")
load("//:buckbuild.bzl", "third_party")

define_glog()

define_xnnpack(third_party)

define_kineto()

# a placeholder for libraries that are not implemented in OSS
cxx_library(
    name = "no-op",
    visibility = ['PUBLIC'],
)

cxx_library(
    name = "rt",
    exported_platform_linker_flags = [
        ("^linux-.*$", ["-lrt"]),
    ],
    visibility = ['PUBLIC'],
)

cxx_library(
    name = "fmt",
    srcs = ['fmt/src/format.cc'],
    deps = [],
    compiler_flags = ['-w', '-Wno-error=format-zero-length', '-Wno-error=vla', '-Wno-incompatible-pointer-types-discards-qualifiers', '-Wno-unused-label', '-Wno-deprecated-declarations', '-Wno-implicit-function-declaration', '-Wno-error', '-Wno-non-pod-varargs', '-Wno-format-security', '-Wno-c++11-narrowing', '-Wno-ignored-attributes', '-Wno-return-std-move', '-Wno-shadow', '-Wno-sign-compare', '-Wno-switch', '-Wno-undef', '-Wno-uninitialized', '-Wno-unknown-pragmas', '-Wno-unknown-warning-option', '-Wno-unused-function', '-Wno-unused-local-typedef', '-Wno-unused-value', '-Wno-unused-variable', '-Wno-register', '-Wno-format', '-Wno-unused-lambda-capture', '-Wno-missing-braces', '-Wno-unused-parameter', '-Wno-unreachable-code', '-Wno-inconsistent-missing-destructor-override', '-Wno-implicit-fallthrough', '-Wno-ignored-qualifiers', '-Wno-pedantic', '-Wno-deprecated-copy', '-Wno-non-virtual-dtor', '-Wno-null-pointer-arithmetic', '-Wno-implicit-const-int-float-conversion', '-Wno-tautological-unsigned-enum-zero-compare', '-Wno-embedded-directive', '-Wno-int-conversion', '-Wno-nonnull', '-Wno-variadic-macros', '-Wno-zero-length-array', '-Wno-missing-prototypes', '-fno-exceptions', '-fno-rtti', '-Wno-braced-scalar-init', '-fvisibility-inlines-hidden'],
    preferred_linkage = "static",
    exported_preprocessor_flags = ['-DFMT_EXCEPTIONS=0'],
    header_namespace = "third_party/fmt",
    public_system_include_directories = ['fmt/include'],
    raw_headers = glob(["fmt/include/fmt/*.h"]),
    soname = "libthird-party_fmt_fmt.$(ext)",
    visibility = ['PUBLIC'],
)

cxx_library(
    name = "pthreadpool",
    srcs = ['pthreadpool/src/legacy-api.c', 'pthreadpool/src/memory.c', 'pthreadpool/src/portable-api.c', 'pthreadpool/src/pthreads.c'],
    deps = [
        ":FXdiv",
        ":pthreadpool_header",
    ],
    compiler_flags = [
        "-w",
        "-Os",
        "-fstack-protector-strong",
        "-fno-delete-null-pointer-checks"
    ],
    headers = {
        'threadpool-atomics.h': 'pthreadpool/src/threadpool-atomics.h',
        'threadpool-common.h': 'pthreadpool/src/threadpool-common.h',
        'threadpool-object.h': 'pthreadpool/src/threadpool-object.h',
        'threadpool-utils.h': 'pthreadpool/src/threadpool-utils.h',
    },
    header_namespace = "",
    preferred_linkage = "static",
    link_whole = False,
    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 = ['-DPTHREADPOOL_USE_FUTEX=0', '-DPTHREADPOOL_USE_GCD=0'],
    reexport_all_header_dependencies = True,
    visibility = ['PUBLIC'],
)

cxx_library(
    name = "pthreadpool_header",
    header_namespace = "",
    exported_headers = {
        "pthreadpool.h": "pthreadpool/include/pthreadpool.h",
    },
    reexport_all_header_dependencies = True,
    visibility = ["PUBLIC"],
)

cxx_library(
    name = "pocketfft_header",
    header_namespace = "",
    exported_headers = {
        "pocketfft_hdronly.h": "pocketfft/pocketfft_hdronly.h",
    },
    reexport_all_header_dependencies = True,
    visibility = ["PUBLIC"],
)

cxx_library(
    name = "FXdiv",
    header_namespace = "",
    exported_headers = {
        "fxdiv.h": "FXdiv/include/fxdiv.h",
    },
    reexport_all_header_dependencies = True,
    visibility = ["PUBLIC"],
)

cxx_library(
    name = "psimd",
    header_namespace = "",
    exported_headers = {
        "psimd.h": "psimd/include/psimd.h",
    },
    preferred_linkage = "static",
    visibility = ["PUBLIC"],
)

cxx_library(
    name = "cpuinfo",
    srcs = [
        "cpuinfo/wrappers/api.c",
        "cpuinfo/wrappers/arm/android/properties.c",
        "cpuinfo/wrappers/arm/cache.c",
        "cpuinfo/wrappers/arm/linux/aarch32-isa.c",
        "cpuinfo/wrappers/arm/linux/aarch64-isa.c",
        "cpuinfo/wrappers/arm/linux/chipset.c",
        "cpuinfo/wrappers/arm/linux/clusters.c",
        "cpuinfo/wrappers/arm/linux/cpuinfo.c",
        "cpuinfo/wrappers/arm/linux/hwcap.c",
        "cpuinfo/wrappers/arm/linux/init.c",
        "cpuinfo/wrappers/arm/linux/midr.c",
        "cpuinfo/wrappers/arm/mach/init.c",
        "cpuinfo/wrappers/arm/uarch.c",
        "cpuinfo/wrappers/cache.c",
        "cpuinfo/wrappers/init.c",
        "cpuinfo/wrappers/linux/cpulist.c",
        "cpuinfo/wrappers/linux/multiline.c",
        "cpuinfo/wrappers/linux/processors.c",
        "cpuinfo/wrappers/linux/smallfile.c",
        "cpuinfo/wrappers/log.c",
        "cpuinfo/wrappers/mach/topology.c",
        "cpuinfo/wrappers/x86/cache/descriptor.c",
        "cpuinfo/wrappers/x86/cache/deterministic.c",
        "cpuinfo/wrappers/x86/cache/init.c",
        "cpuinfo/wrappers/x86/info.c",
        "cpuinfo/wrappers/x86/init.c",
        "cpuinfo/wrappers/x86/isa.c",
        "cpuinfo/wrappers/x86/linux/cpuinfo.c",
        "cpuinfo/wrappers/x86/linux/init.c",
        "cpuinfo/wrappers/x86/mach/init.c",
        "cpuinfo/wrappers/x86/name.c",
        "cpuinfo/wrappers/x86/topology.c",
        "cpuinfo/wrappers/x86/uarch.c",
        "cpuinfo/wrappers/x86/vendor.c",
        "cpuinfo/wrappers/x86/windows/init.c",
    ],
    include_directories = ["cpuinfo/src"],
    public_include_directories = ["cpuinfo/include"],
    raw_headers = glob([
        "cpuinfo/src/**/*.h",
        "cpuinfo/src/**/*.c",
    ]),
    preferred_linkage = "static",
    preprocessor_flags = [
        "-DCPUINFO_LOG_LEVEL=2",
        "-D_GNU_SOURCE=1",
    ],
    visibility = ["PUBLIC"],
    deps = [
        ":clog",
    ],
)

cxx_library(
    name = "clog",
    srcs = [
        "cpuinfo/deps/clog/src/clog.c",
    ],
    raw_headers = glob([
        "cpuinfo/deps/clog/include/*.h",
    ]),
    public_include_directories = [
        "cpuinfo/deps/clog/include/",
    ],
    force_static = True,
    visibility = ["PUBLIC"],
)

cxx_library(
    name = "FP16",
    raw_headers = glob([
        "FP16/include/*.h",
    ]),
    public_include_directories = [
        "FP16/include/",
    ],
    force_static = True,
    visibility = ["PUBLIC"],
)


cxx_library(
    name = "miniz",
    srcs = ["miniz-2.1.0/miniz.c"],
    header_namespace = "",
    exported_headers = {"miniz.h": "miniz-2.1.0/miniz.h"},
    exported_preprocessor_flags = [
        "-DMINIZ_DISABLE_ZIP_READER_CRC32_CHECKS",
    ],
    visibility = ["PUBLIC"],
)

remote_file(
  name = "typing-extensions-download",
  url = "https://files.pythonhosted.org/packages/75/e1/932e06004039dd670c9d5e1df0cd606bf46e29a28e65d5bb28e894ea29c9/typing_extensions-4.2.0-py3-none-any.whl",
  sha1 = "ff0849420e94f425818bff5d0f25e3cdfaba8601",
  out = "typing_extensions-4.2.0-py3-none-any.whl",
)

prebuilt_python_library(
  name = "typing-extensions",
  binary_src = ":typing-extensions-download",
  visibility = ["PUBLIC"],
  deps = [":typing-extensions-download"],
)

remote_file(
  name = "pyyaml-download",
  url = "https://files.pythonhosted.org/packages/12/fc/a4d5a7554e0067677823f7265cb3ae22aed8a238560b5133b58cda252dad/PyYAML-6.0-cp39-cp39-manylinux_2_5_x86_64.manylinux1_x86_64.manylinux_2_12_x86_64.manylinux2010_x86_64.whl",
  sha1 = "11aa9c5fe2d890b6a73212beadc7c8a4265ebc39",
  out = "pyyaml.whl",
)

prebuilt_python_library(
  name = "pyyaml",
  binary_src = ":pyyaml-download",
  visibility = ["PUBLIC"],
  deps = [":pyyaml-download"],
)

cxx_library(
    name = "ruy_lib",
    srcs = glob(
        ["ruy/**/*.cc"],
        exclude = [
            "ruy/ruy/test_*.cc",
            "ruy/ruy/*_test.cc",
            "ruy/example/*.cc",
            "ruy/ruy/profiler/test.cc",
            "ruy/ruy/benchmark.cc",
        ],
    ),
    compiler_flags = ["-Os"],
    preferred_linkage = "static",
    public_include_directories = ["ruy"],
    raw_headers = glob(["ruy/**/*.h"]),
    visibility = [
        "PUBLIC",
    ],
)

cxx_library(
    name = "flatbuffers-api",
    public_include_directories = ["flatbuffers/include"],
    raw_headers = [
        "flatbuffers/include/flatbuffers/base.h",
        "flatbuffers/include/flatbuffers/flatbuffers.h",
        "flatbuffers/include/flatbuffers/stl_emulation.h",
    ],
    preferred_linkage = "static",
    visibility = ["PUBLIC"],
)

cxx_library(
    name = "flatc_library",
    srcs = [
        "flatbuffers/src/code_generators.cpp",
        "flatbuffers/src/flatc.cpp",
        "flatbuffers/src/idl_gen_fbs.cpp",
        "flatbuffers/src/idl_gen_text.cpp",
        "flatbuffers/src/idl_parser.cpp",
        "flatbuffers/src/reflection.cpp",
        "flatbuffers/src/util.cpp",
    ],
    public_include_directories = [
        "flatbuffers/grpc",
        "flatbuffers/include",
    ],
    raw_headers = [
        "flatbuffers/include/flatbuffers/flatc.h",
        "flatbuffers/include/flatbuffers/base.h",
        "flatbuffers/include/flatbuffers/flatbuffers.h",
        "flatbuffers/include/flatbuffers/stl_emulation.h",
        "flatbuffers/include/flatbuffers/code_generators.h",
        "flatbuffers/include/flatbuffers/flexbuffers.h",
        "flatbuffers/include/flatbuffers/hash.h",
        "flatbuffers/include/flatbuffers/idl.h",
        "flatbuffers/include/flatbuffers/minireflect.h",
        "flatbuffers/include/flatbuffers/reflection.h",
        "flatbuffers/include/flatbuffers/reflection_generated.h",
        "flatbuffers/include/flatbuffers/util.h",
    ],
    preferred_linkage = "static",
    visibility = ["PUBLIC"],
)

cxx_binary(
    name = "flatc",
    srcs = [
        "flatbuffers/grpc/src/compiler/cpp_generator.cc",
        "flatbuffers/grpc/src/compiler/go_generator.cc",
        "flatbuffers/grpc/src/compiler/java_generator.cc",
        "flatbuffers/grpc/src/compiler/python_generator.cc",
        "flatbuffers/grpc/src/compiler/swift_generator.cc",
        "flatbuffers/grpc/src/compiler/ts_generator.cc",
        "flatbuffers/src/annotated_binary_text_gen.cpp",
        "flatbuffers/src/binary_annotator.cpp",
        "flatbuffers/src/bfbs_gen_lua.cpp",
        "flatbuffers/src/bfbs_gen_nim.cpp",
        "flatbuffers/src/flatc_main.cpp",
        "flatbuffers/src/idl_gen_binary.cpp",
        "flatbuffers/src/idl_gen_cpp.cpp",
        "flatbuffers/src/idl_gen_csharp.cpp",
        "flatbuffers/src/idl_gen_dart.cpp",
        "flatbuffers/src/idl_gen_go.cpp",
        "flatbuffers/src/idl_gen_grpc.cpp",
        "flatbuffers/src/idl_gen_java.cpp",
        "flatbuffers/src/idl_gen_ts.cpp", # idl_gen_js_ts.cpp in fbsource
        "flatbuffers/src/idl_gen_json_schema.cpp",
        "flatbuffers/src/idl_gen_kotlin.cpp",
        "flatbuffers/src/idl_gen_lobster.cpp",
        "flatbuffers/src/idl_gen_lua.cpp",
        "flatbuffers/src/idl_gen_php.cpp",
        "flatbuffers/src/idl_gen_python.cpp",
        "flatbuffers/src/idl_gen_rust.cpp",
        "flatbuffers/src/idl_gen_swift.cpp",
        "flatbuffers/src/idl_gen_text.cpp",
        "flatbuffers/src/util.cpp",
    ],
    include_directories = [
        "flatbuffers/grpc",
        "flatbuffers/include",
    ],
    raw_headers = [
        "flatbuffers/grpc/src/compiler/cpp_generator.h",
        "flatbuffers/grpc/src/compiler/go_generator.h",
        "flatbuffers/grpc/src/compiler/java_generator.h",
        "flatbuffers/grpc/src/compiler/python_generator.h",
        "flatbuffers/grpc/src/compiler/schema_interface.h",
        "flatbuffers/grpc/src/compiler/swift_generator.h",
        "flatbuffers/grpc/src/compiler/ts_generator.h",
    ],
    visibility = ["PUBLIC"],
    deps = [":flatc_library"],
)

cxx_library(
    name = "gtest_headers",
    exported_preprocessor_flags = [
        "-DGTEST_USE_OWN_TR1_TUPLE=0",
        "-DGTEST_HAS_TR1_TUPLE=0",
        "-D_CRT_DECLARE_NONSTDC_NAMES",
        "-D_CRT_NONSTDC_NO_WARNINGS",
        "-D_CRT_NONSTDC_NO_DEPRECATE",
    ],
    include_directories = [
        "googletest/googletest",
    ],
    public_system_include_directories = [
        "googletest/googletest/include",
    ],
    raw_headers = glob([
        "googletest/googletest/src/**/*.h",
        "googletest/googletest/include/**/*.h",
    ]),
    visibility = [
        "PUBLIC",
    ],
)

cxx_library(
    name = "gtest",
    srcs = [
        "googletest/googletest/src/gtest-all.cc",
        "googletest/googletest/src/gtest_main.cc",
    ],
    include_directories = [
        "googletest/googletest",
    ],
    raw_headers = glob([
        "googletest/googletest/src/**/*.cc",
        "googletest/googletest/src/**/*.h",
    ]),
    visibility = [
        "PUBLIC",
    ],
    exported_deps = [
        ":gtest_headers",
    ],
)

cxx_library(
    name = "gmock",
    srcs = [
        "googletest/googlemock/src/gmock-all.cc",
    ],
    include_directories = [
        "googletest/googlemock",
    ],
    public_system_include_directories = [
        "googletest/googlemock/include",
    ],
    raw_headers = glob([
        "googletest/googlemock/include/**/*.h",
        "googletest/googlemock/src/**/*.cc",
    ]),
    visibility = ["PUBLIC"],
    deps = [
        ":gtest",
    ],
    exported_deps = [
        ":gtest_headers",
    ],
)