diff --git a/third_party/xnnpack.buck.bzl b/third_party/xnnpack.buck.bzl index 4f571377744fe..524e0bd531712 100644 --- a/third_party/xnnpack.buck.bzl +++ b/third_party/xnnpack.buck.bzl @@ -86,7 +86,18 @@ def define_xnnpack(third_party, labels = [], XNNPACK_WINDOWS_AVX512F_ENABLED = F fb_xplat_cxx_library( name = "operators", # srcs have to include HOT_SRCS to be able to build on ARVR - srcs = OPERATOR_SRCS + HOT_SRCS, + srcs = OPERATOR_SRCS + [ + "XNNPACK/src/operators/post-operation.c", + "XNNPACK/src/binary-elementwise-config.c", + "XNNPACK/src/packing.c", + "XNNPACK/src/cache.c", + "XNNPACK/src/indirection.c", + "XNNPACK/src/operator-utils.c", + "XNNPACK/src/normalization.c", + "XNNPACK/src/allocator.c", + "XNNPACK/src/memory.c", + "XNNPACK/src/mutex.c", + ], headers = subdir_glob([ ("XNNPACK/src", "**/*.h"), ]), @@ -185,7 +196,7 @@ def define_xnnpack(third_party, labels = [], XNNPACK_WINDOWS_AVX512F_ENABLED = F fb_xplat_cxx_library( name = "jit_memory", # srcs have to include HOT_SRCS to be able to build on ARVR - srcs = JIT_SRCS, + srcs = JIT_SRCS + HOT_SRCS, headers = subdir_glob([ ("XNNPACK/src", "**/*.h"), ]), @@ -1952,21 +1963,13 @@ def define_xnnpack(third_party, labels = [], XNNPACK_WINDOWS_AVX512F_ENABLED = F "-DXNN_ENABLE_ARM_DOTPROD", ], srcs = [ - "XNNPACK/src/allocator.c", "XNNPACK/src/init.c", "XNNPACK/src/params.c", "XNNPACK/src/operator-run.c", "XNNPACK/src/microparams-init.c", - "XNNPACK/src/binary-elementwise-config.c", - "XNNPACK/src/packing.c", - "XNNPACK/src/indirection.c", - "XNNPACK/src/cache.c", - "XNNPACK/src/mutex.c", - "XNNPACK/src/operator-utils.c", "XNNPACK/src/memory.c", - "XNNPACK/src/hardware-config.c", "XNNPACK/src/x8-lut-config.c", - "XNNPACK/src/normalization.c", + "XNNPACK/src/hardware-config.c", "XNNPACK/src/transpose-config.c", "XNNPACK/src/amalgam/scalar.c", "XNNPACK/src/operators/post-operation.c",