Skip to content

Commit

Permalink
Fix xnnpack link errors (pytorch#101630)
Browse files Browse the repository at this point in the history 
Summary: Setting srcs for all arvr platforms in xnnpack.buck.bzl allows build with arvr/platform010 to pass.

Test Plan: CI

Reviewed By: blchxfm

Differential Revision: D45738660

Pull Request resolved: pytorch#101630
Approved by: https://github.com/digantdesai
  • Loading branch information
@stavhz @pytorchmergebot
stavhz authored and pytorchmergebot committed May 22, 2023
1 parent 2ae87a1 commit 81b0f72
Showing 1 changed file with 12 additions and 48 deletions.
60 changes: 12 additions & 48 deletions third_party/xnnpack.buck.bzl
View file
Original file line number Diff line number Diff line change
Expand Up @@ -257,10 +257,7 @@ def define_xnnpack(third_party, labels = [], XNNPACK_WINDOWS_AVX512F_ENABLED = F

fb_xplat_cxx_library(
name = "ukernels_sse",
srcs = (select({
"DEFAULT": [],
"ovr_config//os:macos-x86_64": PROD_SSE_MICROKERNEL_SRCS,
}) if is_arvr_mode() else []),
srcs = PROD_SSE_MICROKERNEL_SRCS if is_arvr_mode() else [],
headers = subdir_glob([
("XNNPACK/src", "**/*.c"),
("XNNPACK/src", "**/*.h"),
Expand Down Expand Up @@ -340,10 +337,7 @@ def define_xnnpack(third_party, labels = [], XNNPACK_WINDOWS_AVX512F_ENABLED = F

fb_xplat_cxx_library(
name = "ukernels_sse2",
srcs = (select({
"DEFAULT": [],
"ovr_config//os:macos-x86_64": PROD_SSE2_MICROKERNEL_SRCS,
}) if is_arvr_mode() else []),
srcs = PROD_SSE2_MICROKERNEL_SRCS if is_arvr_mode() else [],
headers = subdir_glob([
("XNNPACK/src", "**/*.c"),
("XNNPACK/src", "**/*.h"),
Expand Down Expand Up @@ -425,10 +419,7 @@ def define_xnnpack(third_party, labels = [], XNNPACK_WINDOWS_AVX512F_ENABLED = F

fb_xplat_cxx_library(
name = "ukernels_ssse3",
srcs = (select({
"DEFAULT": [],
"ovr_config//os:macos-x86_64": PROD_SSSE3_MICROKERNEL_SRCS,
}) if is_arvr_mode() else []),
srcs = PROD_SSSE3_MICROKERNEL_SRCS if is_arvr_mode() else [],
headers = subdir_glob([
("XNNPACK/src", "**/*.c"),
("XNNPACK/src", "**/*.h"),
Expand Down Expand Up @@ -510,10 +501,7 @@ def define_xnnpack(third_party, labels = [], XNNPACK_WINDOWS_AVX512F_ENABLED = F

fb_xplat_cxx_library(
name = "ukernels_sse41",
srcs = (select({
"DEFAULT": [],
"ovr_config//os:macos-x86_64": PROD_SSE41_MICROKERNEL_SRCS,
}) if is_arvr_mode() else []),
srcs = PROD_SSE41_MICROKERNEL_SRCS if is_arvr_mode() else [],
headers = subdir_glob([
("XNNPACK/src", "**/*.c"),
("XNNPACK/src", "**/*.h"),
Expand Down Expand Up @@ -595,10 +583,7 @@ def define_xnnpack(third_party, labels = [], XNNPACK_WINDOWS_AVX512F_ENABLED = F

fb_xplat_cxx_library(
name = "ukernels_avx",
srcs = (select({
"DEFAULT": [],
"ovr_config//os:macos-x86_64": PROD_AVX_MICROKERNEL_SRCS,
}) if is_arvr_mode() else []),
srcs = PROD_AVX_MICROKERNEL_SRCS if is_arvr_mode() else [],
headers = subdir_glob([
("XNNPACK/src", "**/*.h"),
("XNNPACK/src", "**/*.c"),
Expand Down Expand Up @@ -680,10 +665,7 @@ def define_xnnpack(third_party, labels = [], XNNPACK_WINDOWS_AVX512F_ENABLED = F

fb_xplat_cxx_library(
name = "ukernels_f16c",
srcs = (select({
"DEFAULT": [],
"ovr_config//os:macos-x86_64": PROD_F16C_MICROKERNEL_SRCS,
}) if is_arvr_mode() else []),
srcs = PROD_F16C_MICROKERNEL_SRCS if is_arvr_mode() else [],
headers = subdir_glob([
("XNNPACK/src", "**/*.h"),
("XNNPACK/src", "**/*.c"),
Expand Down Expand Up @@ -767,10 +749,7 @@ def define_xnnpack(third_party, labels = [], XNNPACK_WINDOWS_AVX512F_ENABLED = F

fb_xplat_cxx_library(
name = "ukernels_xop",
srcs = (select({
"DEFAULT": [],
"ovr_config//os:macos-x86_64": PROD_XOP_MICROKERNEL_SRCS,
}) if is_arvr_mode() else []),
srcs = PROD_XOP_MICROKERNEL_SRCS if is_arvr_mode() else [],
headers = subdir_glob([
("XNNPACK/src", "**/*.h"),
("XNNPACK/src", "**/*.c"),
Expand Down Expand Up @@ -852,10 +831,7 @@ def define_xnnpack(third_party, labels = [], XNNPACK_WINDOWS_AVX512F_ENABLED = F

fb_xplat_cxx_library(
name = "ukernels_fma3",
srcs = (select({
"DEFAULT": [],
"ovr_config//os:macos-x86_64": PROD_FMA3_MICROKERNEL_SRCS,
}) if is_arvr_mode() else []),
srcs = PROD_FMA3_MICROKERNEL_SRCS if is_arvr_mode() else [],
headers = subdir_glob([
("XNNPACK/src", "**/*.h"),
("XNNPACK/src", "**/*.c"),
Expand Down Expand Up @@ -953,10 +929,7 @@ def define_xnnpack(third_party, labels = [], XNNPACK_WINDOWS_AVX512F_ENABLED = F

fb_xplat_cxx_library(
name = "ukernels_avx2",
srcs = (select({
"DEFAULT": [],
"ovr_config//os:macos-x86_64": PROD_AVX2_MICROKERNEL_SRCS,
}) if is_arvr_mode() else []),
srcs = PROD_AVX2_MICROKERNEL_SRCS if is_arvr_mode() else [],
headers = subdir_glob([
("XNNPACK/src", "**/*.c"),
("XNNPACK/src", "**/*.h"),
Expand Down Expand Up @@ -1062,10 +1035,7 @@ def define_xnnpack(third_party, labels = [], XNNPACK_WINDOWS_AVX512F_ENABLED = F

fb_xplat_cxx_library(
name = "ukernels_avx512",
srcs = (select({
"DEFAULT": [],
"ovr_config//os:macos-x86_64": PROD_AVX512F_MICROKERNEL_SRCS,
}) if is_arvr_mode() else []),
srcs = PROD_AVX512F_MICROKERNEL_SRCS if is_arvr_mode() else [],
headers = subdir_glob([
("XNNPACK/src", "**/*.c"),
("XNNPACK/src", "**/*.h"),
Expand Down Expand Up @@ -1109,10 +1079,7 @@ def define_xnnpack(third_party, labels = [], XNNPACK_WINDOWS_AVX512F_ENABLED = F

fb_xplat_cxx_library(
name = "ukernels_avx512vbmi",
srcs = (select({
"DEFAULT": [],
"ovr_config//os:macos-x86_64": PROD_AVX512VBMI_MICROKERNEL_SRCS,
}) if is_arvr_mode() else []),
srcs = PROD_AVX512VBMI_MICROKERNEL_SRCS if is_arvr_mode() else [],
headers = subdir_glob([
("XNNPACK/src", "**/*.c"),
("XNNPACK/src", "**/*.h"),
Expand Down Expand Up @@ -1219,10 +1186,7 @@ def define_xnnpack(third_party, labels = [], XNNPACK_WINDOWS_AVX512F_ENABLED = F

fb_xplat_cxx_library(
name = "ukernels_avx512skx",
srcs = (select({
"DEFAULT": [],
"ovr_config//os:macos-x86_64": PROD_AVX512SKX_MICROKERNEL_SRCS,
}) if is_arvr_mode() else []),
srcs = PROD_AVX512SKX_MICROKERNEL_SRCS if is_arvr_mode() else [],
headers = subdir_glob([
("XNNPACK/src", "**/*.c"),
("XNNPACK/src", "**/*.h"),
Expand Down

0 comments on commit 81b0f72

Please sign in to comment.