mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 12:54:11 +08:00
Enable XNNPACK aarch64 builds (#159762)
Summary: This fixes the build of TorchScript's XNNPACK dependency for our aarch64 device. Thanks to andrewjcg for proposing this fix. Rollback Plan: Reviewed By: andrewjcg Differential Revision: D79497613 Pull Request resolved: https://github.com/pytorch/pytorch/pull/159762 Approved by: https://github.com/frankseide, https://github.com/malfet Co-authored-by: Frank Seide <seide@meta.com>
This commit is contained in:
committed by
PyTorch MergeBot
parent
0de2a45a48
commit
b8ef60b6bc
1
third_party/xnnpack.buck.bzl
vendored
1
third_party/xnnpack.buck.bzl
vendored
@ -2227,6 +2227,7 @@ def define_xnnpack(third_party, labels = [], XNNPACK_WINDOWS_AVX512F_ENABLED = F
|
||||
],
|
||||
# doesn't cover iphonesimulator-x86_64
|
||||
"ovr_config//runtime:arm64-linux-ubuntu-neon": [":arm64_lib"],
|
||||
"ovr_config//runtime:fbcode-arm64": [":arm64_lib"],
|
||||
"ovr_config//runtime:platform010": [":x86_and_x86_64_lib"],
|
||||
}),
|
||||
)
|
||||
|
Reference in New Issue
Block a user