Skip to content

Commit

Permalink
[aarch64] add sleef_arm dependency (pytorch#89988)
Browse files Browse the repository at this point in the history
Reviewed By: kimishpatel, psaab

Differential Revision: D41601965

Pull Request resolved: pytorch#89988
Approved by: https://github.com/soumith
  • Loading branch information
ajaymh authored and pytorchmergebot committed Dec 1, 2022
1 parent 07be48d commit 0ad6715
Showing 1 changed file with 7 additions and 1 deletion.
8 changes: 7 additions & 1 deletion buckbuild.bzl
Original file line number Diff line number Diff line change
Expand Up @@ -147,6 +147,7 @@ THIRD_PARTY_LIBS = {
"rt": ["//xplat/third-party/linker_lib:rt", "//third_party:rt"],
"ruy": ["//third-party/ruy:ruy_xplat_lib", "//third_party:ruy_lib"],
"typing-extensions": ["//third-party/typing-extensions:typing-extensions", "//third_party:typing-extensions"],
"sleef_arm": ["//third-party/sleef:sleef_arm", "//third_party:sleef_arm"],
}

def third_party(name):
Expand Down Expand Up @@ -1930,7 +1931,12 @@ def define_buck_targets(
third_party("glog"),
third_party("XNNPACK"),
third_party("pocketfft"),
],
] + select({
"DEFAULT": [],
"ovr_config//runtime:fbcode-arm64": [
third_party("sleef_arm"),
],
}),
compiler_flags = get_aten_compiler_flags(),
exported_preprocessor_flags = get_aten_preprocessor_flags(),
exported_deps = [
Expand Down

0 comments on commit 0ad6715

Please sign in to comment.