frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

Remove is_arvr_mode() from xnnpack.buck.bzl (#158682)

Browse Source 
Summary:
**Changes**
*   Deleted function import from build definition utilities
    *   Removed `load("//tools/build_defs:fbsource_utils.bzl", "is_arvr_mode")`
*   Replaced is_arvr_mode() function calls with direct references to configuration flags
    *  Changed from `is_arvr_mode()` to `"ovr_config//build_mode:arvr_mode"`
*   Changed conditional expressions to Buck `select()` statements

Test Plan:
Check if CI passes

Rollback Plan:

Differential Revision: D78520947

Pull Request resolved: https://github.com/pytorch/pytorch/pull/158682
Approved by: https://github.com/malfet
This commit is contained in:
Ketan Ambati
2025-07-21 22:49:23 +00:00
committed by PyTorch MergeBot
parent feaa02f9ad
commit f09a484b81
1 changed files with 485 additions and 294 deletions
Download Patch File Download Diff File Expand all files Collapse all files

385
third_party/xnnpack.buck.bzl vendored
View File

@ -1,5 +1,4 @@
load("//tools/build_defs:fb_xplat_cxx_library.bzl", "fb_xplat_cxx_library")
load("//tools/build_defs:fbsource_utils.bzl", "is_arvr_mode")
load("//tools/build_defs:glob_defs.bzl", "subdir_glob")
load("//tools/build_defs:platform_defs.bzl", "ANDROID", "APPLE", "APPLETVOS", "CXX", "IOS", "MACOSX", "WINDOWS")
load(
@ -141,10 +140,13 @@ def define_xnnpack(third_party, labels = [], XNNPACK_WINDOWS_AVX512F_ENABLED = F
fb_xplat_cxx_library(
name = "ukernels_sse",
srcs = select({
"DEFAULT": [],
"ovr_config//build_mode:arvr_mode": select({
"DEFAULT": [],
"ovr_config//cpu:x86_32": prod_srcs_for_arch_wrapper("sse"),
"ovr_config//cpu:x86_64": prod_srcs_for_arch_wrapper("sse"),
}) if is_arvr_mode() else [],
}),
}),
headers = get_xnnpack_headers(),
header_namespace = "",
apple_sdks = (IOS, MACOSX, APPLETVOS),
@ -160,12 +162,15 @@ def define_xnnpack(third_party, labels = [], XNNPACK_WINDOWS_AVX512F_ENABLED = F
],
),
],
platform_srcs = ([
platform_srcs = select({
"DEFAULT": [
(
"x86|x86_64|platform009|platform010",
prod_srcs_for_arch_wrapper("sse"),
),
] if not is_arvr_mode() else []),
],
"ovr_config//build_mode:arvr_mode": [],
}),
fbandroid_link_whole = True,
preferred_linkage = "static",
preprocessor_flags = XNN_COMMON_PREPROCESSOR_FLAGS,
@ -205,10 +210,13 @@ def define_xnnpack(third_party, labels = [], XNNPACK_WINDOWS_AVX512F_ENABLED = F
fb_xplat_cxx_library(
name = "ukernels_sse2",
srcs = select({
"DEFAULT": [],
"ovr_config//build_mode:arvr_mode": select({
"DEFAULT": [],
"ovr_config//cpu:x86_32": prod_srcs_for_arch_wrapper("sse2"),
"ovr_config//cpu:x86_64": prod_srcs_for_arch_wrapper("sse2"),
}) if is_arvr_mode() else [],
}),
}),
headers = get_xnnpack_headers(),
header_namespace = "",
apple_sdks = (IOS, MACOSX, APPLETVOS),
@ -224,12 +232,15 @@ def define_xnnpack(third_party, labels = [], XNNPACK_WINDOWS_AVX512F_ENABLED = F
],
),
],
platform_srcs = ([
platform_srcs = select({
"DEFAULT": [
(
"x86|x86_64|platform009|platform010",
prod_srcs_for_arch_wrapper("sse2"),
),
] if not is_arvr_mode() else []),
],
"ovr_config//build_mode:arvr_mode": [],
}),
fbandroid_link_whole = True,
preferred_linkage = "static",
preprocessor_flags = XNN_COMMON_PREPROCESSOR_FLAGS,
@ -269,10 +280,13 @@ def define_xnnpack(third_party, labels = [], XNNPACK_WINDOWS_AVX512F_ENABLED = F
fb_xplat_cxx_library(
name = "ukernels_ssse3",
srcs = select({
"DEFAULT": [],
"ovr_config//build_mode:arvr_mode": select({
"DEFAULT": [],
"ovr_config//cpu:x86_32": prod_srcs_for_arch_wrapper("ssse3"),
"ovr_config//cpu:x86_64": prod_srcs_for_arch_wrapper("ssse3"),
}) if is_arvr_mode() else [],
}),
}),
headers = get_xnnpack_headers(),
header_namespace = "",
apple_sdks = (IOS, MACOSX, APPLETVOS),
@ -288,12 +302,15 @@ def define_xnnpack(third_party, labels = [], XNNPACK_WINDOWS_AVX512F_ENABLED = F
],
),
],
platform_srcs = ([
platform_srcs = select({
"DEFAULT": [
(
"x86|x86_64|platform009|platform010",
prod_srcs_for_arch_wrapper("ssse3"),
),
] if not is_arvr_mode() else []),
],
"ovr_config//build_mode:arvr_mode": [],
}),
fbandroid_link_whole = True,
preferred_linkage = "static",
preprocessor_flags = XNN_COMMON_PREPROCESSOR_FLAGS,
@ -333,10 +350,13 @@ def define_xnnpack(third_party, labels = [], XNNPACK_WINDOWS_AVX512F_ENABLED = F
fb_xplat_cxx_library(
name = "ukernels_sse41",
srcs = select({
"DEFAULT": [],
"ovr_config//build_mode:arvr_mode": select({
"DEFAULT": [],
"ovr_config//cpu:x86_32": prod_srcs_for_arch_wrapper("sse41"),
"ovr_config//cpu:x86_64": prod_srcs_for_arch_wrapper("sse41"),
}) if is_arvr_mode() else [],
}),
}),
headers = get_xnnpack_headers(),
header_namespace = "",
apple_sdks = (IOS, MACOSX, APPLETVOS),
@ -352,12 +372,15 @@ def define_xnnpack(third_party, labels = [], XNNPACK_WINDOWS_AVX512F_ENABLED = F
],
),
],
platform_srcs = ([
platform_srcs = select({
"DEFAULT": [
(
"x86|x86_64|platform009|platform010",
prod_srcs_for_arch_wrapper("sse41"),
),
] if not is_arvr_mode() else []),
],
"ovr_config//build_mode:arvr_mode": [],
}),
fbandroid_link_whole = True,
preferred_linkage = "static",
preprocessor_flags = XNN_COMMON_PREPROCESSOR_FLAGS,
@ -397,10 +420,13 @@ def define_xnnpack(third_party, labels = [], XNNPACK_WINDOWS_AVX512F_ENABLED = F
fb_xplat_cxx_library(
name = "ukernels_avx",
srcs = select({
"DEFAULT": [],
"ovr_config//build_mode:arvr_mode": select({
"DEFAULT": [],
"ovr_config//cpu:x86_32": prod_srcs_for_arch_wrapper("avx"),
"ovr_config//cpu:x86_64": prod_srcs_for_arch_wrapper("avx"),
}) if is_arvr_mode() else [],
}),
}),
headers = get_xnnpack_headers(),
header_namespace = "",
apple_sdks = (IOS, MACOSX, APPLETVOS),
@ -424,12 +450,15 @@ def define_xnnpack(third_party, labels = [], XNNPACK_WINDOWS_AVX512F_ENABLED = F
],
),
],
platform_srcs = ([
platform_srcs = select({
"DEFAULT": [
(
"x86|x86_64|platform009|platform010",
prod_srcs_for_arch_wrapper("avx"),
),
] if not is_arvr_mode() else []),
],
"ovr_config//build_mode:arvr_mode": [],
}),
fbandroid_link_whole = True,
preferred_linkage = "static",
preprocessor_flags = XNN_COMMON_PREPROCESSOR_FLAGS,
@ -470,10 +499,13 @@ def define_xnnpack(third_party, labels = [], XNNPACK_WINDOWS_AVX512F_ENABLED = F
fb_xplat_cxx_library(
name = "ukernels_avx512vnnigfni",
srcs = select({
"DEFAULT": [],
"ovr_config//build_mode:arvr_mode": select({
"DEFAULT": [],
"ovr_config//cpu:x86_32": prod_srcs_for_arch_wrapper("avx512vnnigfni"),
"ovr_config//cpu:x86_64": prod_srcs_for_arch_wrapper("avx512vnnigfni"),
}) if is_arvr_mode() else [],
}),
}),
headers = get_xnnpack_headers(),
header_namespace = "",
apple_sdks = (IOS, MACOSX, APPLETVOS),
@ -513,12 +545,15 @@ def define_xnnpack(third_party, labels = [], XNNPACK_WINDOWS_AVX512F_ENABLED = F
],
),
],
platform_srcs = ([
platform_srcs = select({
"DEFAULT": [
(
"x86|x86_64|platform009|platform010",
prod_srcs_for_arch_wrapper("avx512vnnigfni"),
),
] if not is_arvr_mode() else []),
],
"ovr_config//build_mode:arvr_mode": [],
}),
preferred_linkage = "static",
preprocessor_flags = XNN_COMMON_PREPROCESSOR_FLAGS,
visibility = ["PUBLIC"],
@ -562,10 +597,13 @@ def define_xnnpack(third_party, labels = [], XNNPACK_WINDOWS_AVX512F_ENABLED = F
fb_xplat_cxx_library(
name = "ukernels_avx512vnni",
srcs = select({
"DEFAULT": [],
"ovr_config//build_mode:arvr_mode": select({
"DEFAULT": [],
"ovr_config//cpu:x86_32": prod_srcs_for_arch_wrapper("avx512vnni"),
"ovr_config//cpu:x86_64": prod_srcs_for_arch_wrapper("avx512vnni"),
}) if is_arvr_mode() else [],
}),
}),
headers = get_xnnpack_headers(),
header_namespace = "",
apple_sdks = (IOS, MACOSX, APPLETVOS),
@ -604,12 +642,15 @@ def define_xnnpack(third_party, labels = [], XNNPACK_WINDOWS_AVX512F_ENABLED = F
],
),
],
platform_srcs = ([
platform_srcs = select({
"DEFAULT": [
(
"x86|x86_64|platform009|platform010",
prod_srcs_for_arch_wrapper("avx512vnni"),
),
] if not is_arvr_mode() else []),
],
"ovr_config//build_mode:arvr_mode": [],
}),
preferred_linkage = "static",
preprocessor_flags = XNN_COMMON_PREPROCESSOR_FLAGS,
exported_preprocessor_flags = [
@ -657,7 +698,10 @@ def define_xnnpack(third_party, labels = [], XNNPACK_WINDOWS_AVX512F_ENABLED = F
fb_xplat_cxx_library(
name = "ukernels_avxvnni",
srcs = prod_srcs_for_arch_wrapper("avxvnni") if is_arvr_mode() else [],
srcs = select({
"DEFAULT": [],
"ovr_config//build_mode:arvr_mode": prod_srcs_for_arch_wrapper("avxvnni"),
}),
headers = get_xnnpack_headers(),
header_namespace = "",
apple_sdks = (IOS, MACOSX, APPLETVOS),
@ -679,12 +723,15 @@ def define_xnnpack(third_party, labels = [], XNNPACK_WINDOWS_AVX512F_ENABLED = F
],
),
],
platform_srcs = ([
platform_srcs = select({
"DEFAULT": [
(
"x86|x86_64|platform009|platform010",
prod_srcs_for_arch_wrapper("avxvnni"),
),
] if not is_arvr_mode() else []),
],
"ovr_config//build_mode:arvr_mode": [],
}),
preferred_linkage = "static",
preprocessor_flags = XNN_COMMON_PREPROCESSOR_FLAGS,
visibility = ["PUBLIC"],
@ -723,10 +770,13 @@ def define_xnnpack(third_party, labels = [], XNNPACK_WINDOWS_AVX512F_ENABLED = F
fb_xplat_cxx_library(
name = "ukernels_f16c",
srcs = select({
"DEFAULT": [],
"ovr_config//build_mode:arvr_mode": select({
"DEFAULT": [],
"ovr_config//cpu:x86_32": prod_srcs_for_arch_wrapper("f16c"),
"ovr_config//cpu:x86_64": prod_srcs_for_arch_wrapper("f16c"),
}) if is_arvr_mode() else [],
}),
}),
headers = get_xnnpack_headers(),
header_namespace = "",
apple_sdks = (IOS, MACOSX, APPLETVOS),
@ -750,12 +800,15 @@ def define_xnnpack(third_party, labels = [], XNNPACK_WINDOWS_AVX512F_ENABLED = F
],
),
],
platform_srcs = ([
platform_srcs = select({
"DEFAULT": [
(
"x86|x86_64|platform009|platform010",
prod_srcs_for_arch_wrapper("f16c"),
),
] if not is_arvr_mode() else []),
],
"ovr_config//build_mode:arvr_mode": [],
}),
platforms = (APPLE, ANDROID, CXX, WINDOWS),
fbandroid_link_whole = True,
preferred_linkage = "static",
@ -798,10 +851,13 @@ def define_xnnpack(third_party, labels = [], XNNPACK_WINDOWS_AVX512F_ENABLED = F
fb_xplat_cxx_library(
name = "ukernels_fma3",
srcs = select({
"DEFAULT": [],
"ovr_config//build_mode:arvr_mode": select({
"DEFAULT": [],
"ovr_config//cpu:x86_32": prod_srcs_for_arch_wrapper("fma3"),
"ovr_config//cpu:x86_64": prod_srcs_for_arch_wrapper("fma3"),
}) if is_arvr_mode() else [],
}),
}),
headers = get_xnnpack_headers(),
header_namespace = "",
apple_sdks = (IOS, MACOSX, APPLETVOS),
@ -828,12 +884,15 @@ def define_xnnpack(third_party, labels = [], XNNPACK_WINDOWS_AVX512F_ENABLED = F
],
),
],
platform_srcs = ([
platform_srcs = select({
"DEFAULT": [
(
"x86|x86_64|platform009|platform010",
prod_srcs_for_arch_wrapper("fma3"),
),
] if not is_arvr_mode() else []),
],
"ovr_config//build_mode:arvr_mode": [],
}),
fbandroid_link_whole = True,
preferred_linkage = "static",
preprocessor_flags = XNN_COMMON_PREPROCESSOR_FLAGS,
@ -888,10 +947,13 @@ def define_xnnpack(third_party, labels = [], XNNPACK_WINDOWS_AVX512F_ENABLED = F
fb_xplat_cxx_library(
name = "ukernels_avx2",
srcs = select({
"DEFAULT": [],
"ovr_config//build_mode:arvr_mode": select({
"DEFAULT": [],
"ovr_config//cpu:x86_32": prod_srcs_for_arch_wrapper("avx2"),
"ovr_config//cpu:x86_64": prod_srcs_for_arch_wrapper("avx2"),
}) if is_arvr_mode() else [],
}),
}),
headers = get_xnnpack_headers(),
header_namespace = "",
apple_sdks = (IOS, MACOSX, APPLETVOS),
@ -921,12 +983,15 @@ def define_xnnpack(third_party, labels = [], XNNPACK_WINDOWS_AVX512F_ENABLED = F
],
),
],
platform_srcs = ([
platform_srcs = select({
"DEFAULT": [
(
"x86|x86_64|platform009|platform010",
prod_srcs_for_arch_wrapper("avx2"),
),
] if not is_arvr_mode() else []),
],
"ovr_config//build_mode:arvr_mode": [],
}),
fbandroid_link_whole = True,
preferred_linkage = "static",
preprocessor_flags = XNN_COMMON_PREPROCESSOR_FLAGS,
@ -988,10 +1053,13 @@ def define_xnnpack(third_party, labels = [], XNNPACK_WINDOWS_AVX512F_ENABLED = F
fb_xplat_cxx_library(
name = "ukernels_avx512",
srcs = select({
"DEFAULT": [],
"ovr_config//build_mode:arvr_mode": select({
"DEFAULT": [],
"ovr_config//cpu:x86_32": prod_srcs_for_arch_wrapper("avx512f"),
"ovr_config//cpu:x86_64": prod_srcs_for_arch_wrapper("avx512f"),
}) if is_arvr_mode() else [],
}),
}),
headers = get_xnnpack_headers(),
header_namespace = "",
apple_sdks = (IOS, MACOSX, APPLETVOS),
@ -1015,12 +1083,15 @@ def define_xnnpack(third_party, labels = [], XNNPACK_WINDOWS_AVX512F_ENABLED = F
],
),
],
platform_srcs = ([
platform_srcs = select({
"DEFAULT": [
(
"x86|x86_64|platform009|platform010",
prod_srcs_for_arch_wrapper("avx512f"),
),
] if not is_arvr_mode() else []),
],
"ovr_config//build_mode:arvr_mode": [],
}),
fbandroid_link_whole = True,
preferred_linkage = "static",
preprocessor_flags = XNN_COMMON_PREPROCESSOR_FLAGS,
@ -1033,10 +1104,13 @@ def define_xnnpack(third_party, labels = [], XNNPACK_WINDOWS_AVX512F_ENABLED = F
fb_xplat_cxx_library(
name = "ukernels_avx512vbmi",
srcs = select({
"DEFAULT": [],
"ovr_config//build_mode:arvr_mode": select({
"DEFAULT": [],
"ovr_config//cpu:x86_32": prod_srcs_for_arch_wrapper("avx512vbmi"),
"ovr_config//cpu:x86_64": prod_srcs_for_arch_wrapper("avx512vbmi"),
}) if is_arvr_mode() else [],
}),
}),
headers = get_xnnpack_headers(),
header_namespace = "",
apple_sdks = (IOS, MACOSX, APPLETVOS),
@ -1075,12 +1149,15 @@ def define_xnnpack(third_party, labels = [], XNNPACK_WINDOWS_AVX512F_ENABLED = F
],
),
],
platform_srcs = ([
platform_srcs = select({
"DEFAULT": [
(
"x86|x86_64|platform009|platform010",
prod_srcs_for_arch_wrapper("avx512vbmi"),
),
] if not is_arvr_mode() else []),
],
"ovr_config//build_mode:arvr_mode": [],
}),
fbandroid_link_whole = True,
preferred_linkage = "static",
preprocessor_flags = XNN_COMMON_PREPROCESSOR_FLAGS,
@ -1135,10 +1212,13 @@ def define_xnnpack(third_party, labels = [], XNNPACK_WINDOWS_AVX512F_ENABLED = F
fb_xplat_cxx_library(
name = "ukernels_avx512skx",
srcs = select({
"DEFAULT": [],
"ovr_config//build_mode:arvr_mode": select({
"DEFAULT": [],
"ovr_config//cpu:x86_32": prod_srcs_for_arch_wrapper("avx512skx"),
"ovr_config//cpu:x86_64": prod_srcs_for_arch_wrapper("avx512skx"),
}) if is_arvr_mode() else [],
}),
}),
headers = get_xnnpack_headers(),
header_namespace = "",
apple_sdks = (IOS, MACOSX, APPLETVOS),
@ -1174,12 +1254,15 @@ def define_xnnpack(third_party, labels = [], XNNPACK_WINDOWS_AVX512F_ENABLED = F
],
),
],
platform_srcs = ([
platform_srcs = select({
"DEFAULT": [
(
"x86|x86_64|platform009|platform010",
prod_srcs_for_arch_wrapper("avx512skx"),
),
] if not is_arvr_mode() else []),
],
"ovr_config//build_mode:arvr_mode": [],
}),
fbandroid_link_whole = True,
preferred_linkage = "static",
preprocessor_flags = XNN_COMMON_PREPROCESSOR_FLAGS,
@ -1254,9 +1337,12 @@ def define_xnnpack(third_party, labels = [], XNNPACK_WINDOWS_AVX512F_ENABLED = F
fb_xplat_cxx_library(
name = "ukernels_armsimd32",
srcs = select({
"DEFAULT": [],
"ovr_config//build_mode:arvr_mode": select({
"DEFAULT": [],
"ovr_config//cpu:arm32": prod_srcs_for_arch_wrapper("armsimd32"),
}) if is_arvr_mode() else [],
}),
}),
headers = get_xnnpack_headers(),
header_namespace = "",
apple_sdks = (IOS, MACOSX, APPLETVOS),
@ -1277,12 +1363,15 @@ def define_xnnpack(third_party, labels = [], XNNPACK_WINDOWS_AVX512F_ENABLED = F
],
),
],
platform_srcs = [
platform_srcs = select({
"DEFAULT": [
(
"(aarch32|arm32|armv7)",
prod_srcs_for_arch_wrapper("armsimd32"),
),
] if not is_arvr_mode() else [],
],
"ovr_config//build_mode:arvr_mode": [],
}),
fbandroid_link_whole = True,
preferred_linkage = "static",
preprocessor_flags = XNN_COMMON_PREPROCESSOR_FLAGS,
@ -1295,10 +1384,13 @@ def define_xnnpack(third_party, labels = [], XNNPACK_WINDOWS_AVX512F_ENABLED = F
fb_xplat_cxx_library(
name = "ukernels_neon",
srcs = select({
"DEFAULT": [],
"ovr_config//build_mode:arvr_mode": select({
"DEFAULT": [],
"ovr_config//cpu:arm32": prod_srcs_for_arch_wrapper("neon"),
"ovr_config//cpu:arm64": prod_srcs_for_arch_wrapper("neon"),
}) if is_arvr_mode() else [],
}),
}),
headers = get_xnnpack_headers(),
header_namespace = "",
apple_sdks = (IOS, MACOSX, APPLETVOS),
@ -1323,7 +1415,8 @@ def define_xnnpack(third_party, labels = [], XNNPACK_WINDOWS_AVX512F_ENABLED = F
],
),
],
platform_srcs = [
platform_srcs = select({
"DEFAULT": [
(
"(aarch32|arm32|armv7)",
prod_srcs_for_arch_wrapper("neon"),
@ -1332,7 +1425,9 @@ def define_xnnpack(third_party, labels = [], XNNPACK_WINDOWS_AVX512F_ENABLED = F
"(aarch64|arm64)",
prod_srcs_for_arch_wrapper("neon"),
),
] if not is_arvr_mode() else [],
],
"ovr_config//build_mode:arvr_mode": [],
}),
fbandroid_link_whole = True,
preferred_linkage = "static",
preprocessor_flags = XNN_COMMON_PREPROCESSOR_FLAGS,
@ -1345,21 +1440,27 @@ def define_xnnpack(third_party, labels = [], XNNPACK_WINDOWS_AVX512F_ENABLED = F
fb_xplat_cxx_library(
name = "ukernels_neon_aarch64",
srcs = select({
"DEFAULT": [],
"ovr_config//build_mode:arvr_mode": select({
"DEFAULT": [],
"ovr_config//cpu:arm64": prod_srcs_for_arch_wrapper("neon_aarch64"),
}) if is_arvr_mode() else [],
}),
}),
headers = get_xnnpack_headers(),
header_namespace = "",
apple_sdks = (IOS, MACOSX, APPLETVOS),
compiler_flags = [
"-O2",
],
platform_srcs = [
platform_srcs = select({
"DEFAULT": [
(
"(aarch64|arm64)",
prod_srcs_for_arch_wrapper("neon_aarch64"),
),
] if not is_arvr_mode() else [],
],
"ovr_config//build_mode:arvr_mode": [],
}),
labels = labels,
fbandroid_link_whole = True,
preferred_linkage = "static",
@ -1373,9 +1474,12 @@ def define_xnnpack(third_party, labels = [], XNNPACK_WINDOWS_AVX512F_ENABLED = F
fb_xplat_cxx_library(
name = "ukernels_neon_fma",
srcs = select({
"DEFAULT": [],
"ovr_config//build_mode:arvr_mode": select({
"DEFAULT": [],
"ovr_config//cpu:arm32": prod_srcs_for_arch_wrapper("neonfma"),
}) if is_arvr_mode() else [],
}),
}),
headers = get_xnnpack_headers(),
header_namespace = "",
apple_sdks = (IOS, MACOSX, APPLETVOS),
@ -1407,12 +1511,15 @@ def define_xnnpack(third_party, labels = [], XNNPACK_WINDOWS_AVX512F_ENABLED = F
],
),
],
platform_srcs = [
platform_srcs = select({
"DEFAULT": [
(
"(aarch32|arm32|armv7)",
prod_srcs_for_arch_wrapper("neonfma"),
),
] if not is_arvr_mode() else [],
],
"ovr_config//build_mode:arvr_mode": [],
}),
fbandroid_link_whole = True,
preferred_linkage = "static",
preprocessor_flags = XNN_COMMON_PREPROCESSOR_FLAGS,
@ -1425,9 +1532,12 @@ def define_xnnpack(third_party, labels = [], XNNPACK_WINDOWS_AVX512F_ENABLED = F
fb_xplat_cxx_library(
name = "ukernels_neonfma_aarch64",
srcs = select({
"DEFAULT": [],
"ovr_config//build_mode:arvr_mode": select({
"DEFAULT": [],
"ovr_config//cpu:arm64": prod_srcs_for_arch_wrapper("neonfma") + prod_srcs_for_arch_wrapper("neonfma_aarch64"),
}) if is_arvr_mode() else [],
}),
}),
headers = get_xnnpack_headers(),
header_namespace = "",
apple_sdks = (IOS, MACOSX, APPLETVOS),
@ -1435,12 +1545,15 @@ def define_xnnpack(third_party, labels = [], XNNPACK_WINDOWS_AVX512F_ENABLED = F
"-O2",
],
labels = labels,
platform_srcs = [
platform_srcs = select({
"DEFAULT": [
(
"(arm64|aarch64)$",
prod_srcs_for_arch_wrapper("neonfma") + prod_srcs_for_arch_wrapper("neonfma_aarch64"),
),
] if not is_arvr_mode() else [],
],
"ovr_config//build_mode:arvr_mode": [],
}),
platforms = (APPLE, ANDROID, CXX, WINDOWS),
fbandroid_link_whole = True,
preferred_linkage = "static",
@ -1454,10 +1567,13 @@ def define_xnnpack(third_party, labels = [], XNNPACK_WINDOWS_AVX512F_ENABLED = F
fb_xplat_cxx_library(
name = "ukernels_fp16arith",
srcs = select({
"DEFAULT": [],
"ovr_config//build_mode:arvr_mode": select({
"DEFAULT": [],
"ovr_config//cpu:arm32": prod_srcs_for_arch_wrapper("fp16arith"),
"ovr_config//cpu:arm64": prod_srcs_for_arch_wrapper("fp16arith"),
}) if is_arvr_mode() else [],
}),
}),
headers = get_xnnpack_headers(),
header_namespace = "",
apple_sdks = (IOS, MACOSX, APPLETVOS),
@ -1504,7 +1620,8 @@ def define_xnnpack(third_party, labels = [], XNNPACK_WINDOWS_AVX512F_ENABLED = F
],
)
],
platform_srcs = [
platform_srcs = select({
"DEFAULT": [
(
"(aarch32|arm32|armv7)",
prod_srcs_for_arch_wrapper("fp16arith"),
@ -1513,7 +1630,9 @@ def define_xnnpack(third_party, labels = [], XNNPACK_WINDOWS_AVX512F_ENABLED = F
"(aarch64|arm64)",
prod_srcs_for_arch_wrapper("fp16arith") + prod_srcs_for_arch_wrapper("fp16arith_aarch64"),
),
] if not is_arvr_mode() else [],
],
"ovr_config//build_mode:arvr_mode": [],
}),
fbandroid_link_whole = True,
preferred_linkage = "static",
preprocessor_flags = XNN_COMMON_PREPROCESSOR_FLAGS,
@ -1526,10 +1645,13 @@ def define_xnnpack(third_party, labels = [], XNNPACK_WINDOWS_AVX512F_ENABLED = F
fb_xplat_cxx_library(
name = "ukernels_neon_fp16",
srcs = select({
"DEFAULT": [],
"ovr_config//build_mode:arvr_mode": select({
"DEFAULT": [],
"ovr_config//cpu:arm32": prod_srcs_for_arch_wrapper("neonfp16"),
"ovr_config//cpu:arm64": prod_srcs_for_arch_wrapper("neonfp16"),
}) if is_arvr_mode() else [],
}),
}),
headers = get_xnnpack_headers(),
header_namespace = "",
apple_sdks = (IOS, MACOSX, APPLETVOS),
@ -1554,7 +1676,8 @@ def define_xnnpack(third_party, labels = [], XNNPACK_WINDOWS_AVX512F_ENABLED = F
],
),
],
platform_srcs = [
platform_srcs = select({
"DEFAULT": [
(
"(aarch32|arm32|armv7)$",
prod_srcs_for_arch_wrapper("neonfp16"),
@ -1563,7 +1686,9 @@ def define_xnnpack(third_party, labels = [], XNNPACK_WINDOWS_AVX512F_ENABLED = F
"(arm64|aarch64)",
prod_srcs_for_arch_wrapper("neonfp16"),
),
] if not is_arvr_mode() else [],
],
"ovr_config//build_mode:arvr_mode": [],
}),
fbandroid_link_whole = True,
preferred_linkage = "static",
preprocessor_flags = XNN_COMMON_PREPROCESSOR_FLAGS,
@ -1576,10 +1701,13 @@ def define_xnnpack(third_party, labels = [], XNNPACK_WINDOWS_AVX512F_ENABLED = F
fb_xplat_cxx_library(
name = "ukernels_neon_v8",
srcs = select({
"DEFAULT": [],
"ovr_config//build_mode:arvr_mode": select({
"DEFAULT": [],
"ovr_config//cpu:arm32": prod_srcs_for_arch_wrapper("neonv8"),
"ovr_config//cpu:arm64": prod_srcs_for_arch_wrapper("neonv8"),
}) if is_arvr_mode() else [],
}),
}),
headers = get_xnnpack_headers(),
header_namespace = "",
apple_sdks = (IOS, MACOSX, APPLETVOS),
@ -1618,7 +1746,8 @@ def define_xnnpack(third_party, labels = [], XNNPACK_WINDOWS_AVX512F_ENABLED = F
],
),
],
platform_srcs = [
platform_srcs = select({
"DEFAULT": [
(
"(aarch32|arm32|armv7)$",
prod_srcs_for_arch_wrapper("neonv8"),
@ -1627,7 +1756,9 @@ def define_xnnpack(third_party, labels = [], XNNPACK_WINDOWS_AVX512F_ENABLED = F
"(arm64|aarch64)",
prod_srcs_for_arch_wrapper("neonv8"),
),
] if not is_arvr_mode() else [],
],
"ovr_config//build_mode:arvr_mode": [],
}),
fbandroid_link_whole = True,
preferred_linkage = "static",
preprocessor_flags = XNN_COMMON_PREPROCESSOR_FLAGS,
@ -1640,9 +1771,12 @@ def define_xnnpack(third_party, labels = [], XNNPACK_WINDOWS_AVX512F_ENABLED = F
fb_xplat_cxx_library(
name = "ukernels_neon_dot",
srcs = select({
"DEFAULT": [],
"ovr_config//build_mode:arvr_mode": select({
"DEFAULT": [],
"ovr_config//cpu:arm32": prod_srcs_for_arch_wrapper("neondot"),
}) if is_arvr_mode() else [],
}),
}),
headers = get_xnnpack_headers(),
header_namespace = "",
apple_sdks = (IOS, MACOSX, APPLETVOS),
@ -1667,12 +1801,15 @@ def define_xnnpack(third_party, labels = [], XNNPACK_WINDOWS_AVX512F_ENABLED = F
],
),
],
platform_srcs = [
platform_srcs = select({
"DEFAULT": [
(
"(aarch32|arm32|armv7)",
prod_srcs_for_arch_wrapper("neondot"),
),
] if not is_arvr_mode() else [],
],
"ovr_config//build_mode:arvr_mode": [],
}),
fbandroid_link_whole = True,
preferred_linkage = "static",
preprocessor_flags = XNN_COMMON_PREPROCESSOR_FLAGS,
@ -1685,9 +1822,12 @@ def define_xnnpack(third_party, labels = [], XNNPACK_WINDOWS_AVX512F_ENABLED = F
fb_xplat_cxx_library(
name = "ukernels_neon_dot_aarch64",
srcs = select({
"DEFAULT": [],
"ovr_config//build_mode:arvr_mode": select({
"DEFAULT": [],
"ovr_config//cpu:arm64": prod_srcs_for_arch_wrapper("neondot") + prod_srcs_for_arch_wrapper("neondot_aarch64"),
}) if is_arvr_mode() else [],
}),
}),
headers = get_xnnpack_headers(),
header_namespace = "",
apple_sdks = (IOS, MACOSX, APPLETVOS),
@ -1706,12 +1846,15 @@ def define_xnnpack(third_party, labels = [], XNNPACK_WINDOWS_AVX512F_ENABLED = F
],
),
],
platform_srcs = [
platform_srcs = select({
"DEFAULT": [
(
"(aarch64|arm64)",
prod_srcs_for_arch_wrapper("neondot") + prod_srcs_for_arch_wrapper("neondot_aarch64"),
),
] if not is_arvr_mode() else [],
],
"ovr_config//build_mode:arvr_mode": [],
}),
fbandroid_link_whole = True,
preferred_linkage = "static",
preprocessor_flags = XNN_COMMON_PREPROCESSOR_FLAGS,
@ -1724,9 +1867,12 @@ def define_xnnpack(third_party, labels = [], XNNPACK_WINDOWS_AVX512F_ENABLED = F
fb_xplat_cxx_library(
name = "ukernels_neon_dot_fp16arith",
srcs = select({
"DEFAULT": [],
"ovr_config//build_mode:arvr_mode": select({
"DEFAULT": [],
"ovr_config//cpu:arm32": prod_srcs_for_arch_wrapper("neondotfp16arith"),
}) if is_arvr_mode() else [],
}),
}),
headers = get_xnnpack_headers(),
header_namespace = "",
apple_sdks = (IOS, MACOSX, APPLETVOS),
@ -1750,12 +1896,15 @@ def define_xnnpack(third_party, labels = [], XNNPACK_WINDOWS_AVX512F_ENABLED = F
],
),
],
platform_srcs = [
platform_srcs = select({
"DEFAULT": [
(
"(aarch32|arm32|armv7)",
prod_srcs_for_arch_wrapper("neondotfp16arith"),
),
] if not is_arvr_mode() else [],
],
"ovr_config//build_mode:arvr_mode": [],
}),
labels = labels,
fbandroid_link_whole = True,
preferred_linkage = "static",
@ -1769,9 +1918,12 @@ def define_xnnpack(third_party, labels = [], XNNPACK_WINDOWS_AVX512F_ENABLED = F
fb_xplat_cxx_library(
name = "ukernels_neon_dot_fp16arith_aarch64",
srcs = select({
"DEFAULT": [],
"ovr_config//build_mode:arvr_mode": select({
"DEFAULT": [],
"ovr_config//cpu:arm64": prod_srcs_for_arch_wrapper("neondotfp16arith") + prod_srcs_for_arch_wrapper("neondotfp16arith_aarch64"),
}) if is_arvr_mode() else [],
}),
}),
headers = get_xnnpack_headers(),
header_namespace = "",
apple_sdks = (IOS, MACOSX, APPLETVOS),
@ -1791,12 +1943,15 @@ def define_xnnpack(third_party, labels = [], XNNPACK_WINDOWS_AVX512F_ENABLED = F
],
),
],
platform_srcs = [
platform_srcs = select({
"DEFAULT": [
(
"(aarch64|arm64)",
prod_srcs_for_arch_wrapper("neondotfp16arith") + prod_srcs_for_arch_wrapper("neondotfp16arith_aarch64"),
),
] if not is_arvr_mode() else [],
],
"ovr_config//build_mode:arvr_mode": [],
}),
labels = labels,
fbandroid_link_whole = True,
preferred_linkage = "static",
@ -1810,9 +1965,12 @@ def define_xnnpack(third_party, labels = [], XNNPACK_WINDOWS_AVX512F_ENABLED = F
fb_xplat_cxx_library(
name = "ukernels_neon_fp16arith",
srcs = select({
"DEFAULT": [],
"ovr_config//build_mode:arvr_mode": select({
"DEFAULT": [],
"ovr_config//cpu:arm32": prod_srcs_for_arch_wrapper("neonfp16arith"),
}) if is_arvr_mode() else [],
}),
}),
headers = get_xnnpack_headers(),
header_namespace = "",
apple_sdks = (IOS, MACOSX, APPLETVOS),
@ -1837,12 +1995,15 @@ def define_xnnpack(third_party, labels = [], XNNPACK_WINDOWS_AVX512F_ENABLED = F
],
),
],
platform_srcs = [
platform_srcs = select({
"DEFAULT": [
(
"(aarch32|arm32|armv7)",
prod_srcs_for_arch_wrapper("neonfp16arith"),
),
] if not is_arvr_mode() else [],
],
"ovr_config//build_mode:arvr_mode": [],
}),
fbandroid_link_whole = True,
preferred_linkage = "static",
preprocessor_flags = XNN_COMMON_PREPROCESSOR_FLAGS,
@ -1855,9 +2016,12 @@ def define_xnnpack(third_party, labels = [], XNNPACK_WINDOWS_AVX512F_ENABLED = F
fb_xplat_cxx_library(
name = "ukernels_neon_fp16arith_aarch64",
srcs = select({
"DEFAULT": [],
"ovr_config//build_mode:arvr_mode": select({
"DEFAULT": [],
"ovr_config//cpu:arm64": prod_srcs_for_arch_wrapper("neonfp16arith") + prod_srcs_for_arch_wrapper("neonfp16arith_aarch64"),
}) if is_arvr_mode() else [],
}),
}),
headers = get_xnnpack_headers(),
header_namespace = "",
apple_sdks = (IOS, MACOSX, APPLETVOS),
@ -1876,12 +2040,15 @@ def define_xnnpack(third_party, labels = [], XNNPACK_WINDOWS_AVX512F_ENABLED = F
],
),
],
platform_srcs = [
platform_srcs = select({
"DEFAULT": [
(
"(aarch64|arm64)",
prod_srcs_for_arch_wrapper("neonfp16arith") + prod_srcs_for_arch_wrapper("neonfp16arith_aarch64"),
),
] if not is_arvr_mode() else [],
],
"ovr_config//build_mode:arvr_mode": [],
}),
fbandroid_link_whole = True,
preferred_linkage = "static",
preprocessor_flags = XNN_COMMON_PREPROCESSOR_FLAGS,
@ -1894,10 +2061,13 @@ def define_xnnpack(third_party, labels = [], XNNPACK_WINDOWS_AVX512F_ENABLED = F
fb_xplat_cxx_library(
name = "ukernels_neonfma_i8mm",
srcs = select({
"DEFAULT": [],
"ovr_config//build_mode:arvr_mode": select({
"DEFAULT": [],
"ovr_config//cpu:arm32": prod_srcs_for_arch_wrapper("neonfma_i8mm"),
"ovr_config//cpu:arm64": prod_srcs_for_arch_wrapper("neonfma_i8mm"),
}) if is_arvr_mode() else [],
}),
}),
headers = get_xnnpack_headers(),
header_namespace = "",
apple_sdks = (IOS, MACOSX, APPLETVOS),
@ -1931,7 +2101,8 @@ def define_xnnpack(third_party, labels = [], XNNPACK_WINDOWS_AVX512F_ENABLED = F
],
),
],
platform_srcs = [
platform_srcs = select({
"DEFAULT": [
(
"(aarch32|arm32|armv7)$",
prod_srcs_for_arch_wrapper("neonfma_i8mm"),
@ -1940,7 +2111,9 @@ def define_xnnpack(third_party, labels = [], XNNPACK_WINDOWS_AVX512F_ENABLED = F
"(arm64|aarch64)",
prod_srcs_for_arch_wrapper("neonfma_i8mm"),
),
] if not is_arvr_mode() else [],
],
"ovr_config//build_mode:arvr_mode": [],
}),
platforms = (APPLE, ANDROID, CXX, WINDOWS),
fbandroid_link_whole = True,
preferred_linkage = "static",
@ -1954,9 +2127,12 @@ def define_xnnpack(third_party, labels = [], XNNPACK_WINDOWS_AVX512F_ENABLED = F
fb_xplat_cxx_library(
name = "ukernels_neoni8mm",
srcs = select({
"DEFAULT": [],
"ovr_config//build_mode:arvr_mode": select({
"DEFAULT": [],
"ovr_config//cpu:arm64": prod_srcs_for_arch_wrapper("neoni8mm"),
}) if is_arvr_mode() else [],
}),
}),
headers = get_xnnpack_headers(),
header_namespace = "",
apple_sdks = (IOS, MACOSX, APPLETVOS),
@ -1977,12 +2153,15 @@ def define_xnnpack(third_party, labels = [], XNNPACK_WINDOWS_AVX512F_ENABLED = F
],
),
],
platform_srcs = [
platform_srcs = select({
"DEFAULT": [
(
"(arm64|aarch64)",
prod_srcs_for_arch_wrapper("neoni8mm"),
),
] if not is_arvr_mode() else [],
],
"ovr_config//build_mode:arvr_mode": [],
}),
platforms = (APPLE, ANDROID, CXX, WINDOWS),
fbandroid_link_whole = True,
preferred_linkage = "static",
@ -1996,9 +2175,12 @@ def define_xnnpack(third_party, labels = [], XNNPACK_WINDOWS_AVX512F_ENABLED = F
fb_xplat_cxx_library(
name = "ukernels_asm_aarch32",
srcs = select({
"DEFAULT": [],
"ovr_config//build_mode:arvr_mode": select({
"DEFAULT": [],
"ovr_config//cpu:arm32": prod_srcs_for_arch_wrapper("aarch32"),
}) if is_arvr_mode() else [],
}),
}),
headers = subdir_glob([
("XNNPACK/src", "xnnpack/assembly.h"),
("XNNPACK/src", "**/*.S"),
@ -2026,12 +2208,15 @@ def define_xnnpack(third_party, labels = [], XNNPACK_WINDOWS_AVX512F_ENABLED = F
],
),
],
platform_srcs = [
platform_srcs = select({
"DEFAULT": [
(
"(aarch32|arm32|armv7)",
prod_srcs_for_arch_wrapper("aarch32"),
),
] if not is_arvr_mode() else [],
],
"ovr_config//build_mode:arvr_mode": [],
}),
platforms = (APPLE, ANDROID, CXX, WINDOWS),
fbandroid_link_whole = True,
preferred_linkage = "static",
@ -2045,9 +2230,12 @@ def define_xnnpack(third_party, labels = [], XNNPACK_WINDOWS_AVX512F_ENABLED = F
fb_xplat_cxx_library(
name = "ukernels_asm_aarch64",
srcs = select({
"DEFAULT": [],
"ovr_config//build_mode:arvr_mode": select({
"DEFAULT": [],
"ovr_config//cpu:arm64": prod_srcs_for_arch_wrapper("aarch64"),
}) if is_arvr_mode() else [],
}),
}),
headers = subdir_glob([
("XNNPACK/src", "xnnpack/assembly.h"),
("XNNPACK/src", "**/*.S"),
@ -2071,12 +2259,15 @@ def define_xnnpack(third_party, labels = [], XNNPACK_WINDOWS_AVX512F_ENABLED = F
],
),
],
platform_srcs = [
platform_srcs = select({
"DEFAULT": [
(
"(aarch64|arm64)",
prod_srcs_for_arch_wrapper("aarch64"),
),
] if not is_arvr_mode() else [],
],
"ovr_config//build_mode:arvr_mode": [],
}),
fbandroid_link_whole = True,
preferred_linkage = "static",
preprocessor_flags = XNN_COMMON_PREPROCESSOR_FLAGS,