mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 12:54:11 +08:00
cca34be584
Updating XNNPACK Version to 4ea82e595b36106653175dcb04b2aa532660d0d8 submodule update Pull Request resolved: https://github.com/pytorch/pytorch/pull/139913 Approved by: https://github.com/digantdesai, https://github.com/huydhn
34 lines
1.1 KiB
Python
34 lines
1.1 KiB
Python
load(
|
|
"//xplat/third-party/XNNPACK/XNNPACK:build_srcs.bzl",
|
|
_LOGGING_SRCS = "LOGGING_SRCS",
|
|
_OPERATOR_SRCS = "OPERATOR_SRCS",
|
|
_SUBGRAPH_SRCS = "SUBGRAPH_SRCS",
|
|
_TABLE_SRCS = "TABLE_SRCS",
|
|
_XNNPACK_SRCS = "XNNPACK_SRCS",
|
|
)
|
|
load("//xplat/third-party/XNNPACK/XNNPACK/gen:microkernels.bzl", "prod_srcs_for_arch")
|
|
load("//tools/build_defs:glob_defs.bzl", "subdir_glob")
|
|
|
|
def define_xnnpack_build_src(xnnpack_build_src):
|
|
return ["XNNPACK/{}".format(src) for src in xnnpack_build_src]
|
|
|
|
def prod_srcs_for_arch_wrapper(arch):
|
|
prod_srcs = prod_srcs_for_arch(arch)
|
|
return define_xnnpack_build_src(prod_srcs)
|
|
|
|
def get_xnnpack_headers():
|
|
src_headers = subdir_glob([
|
|
("XNNPACK/src", "**/*.h"),
|
|
])
|
|
include_headers = subdir_glob([
|
|
("XNNPACK/include", "*.h"),
|
|
])
|
|
|
|
return src_headers | include_headers
|
|
|
|
OPERATOR_SRCS = define_xnnpack_build_src(_OPERATOR_SRCS)
|
|
SUBGRAPH_SRCS = define_xnnpack_build_src(_SUBGRAPH_SRCS)
|
|
TABLE_SRCS = define_xnnpack_build_src(_TABLE_SRCS)
|
|
XNNPACK_SRCS = define_xnnpack_build_src(_XNNPACK_SRCS)
|
|
LOGGING_SRCS = define_xnnpack_build_src(_LOGGING_SRCS)
|