Skip to content

Commit f09bd55

Browse files
JacobSzwejbkaDark Knight
andauthored
Revert D95224222 (pytorch#18020)
Summary: This diff reverts D95224222 Depends on D95837397 (The context such as a Sandcastle job, Task, SEV, etc. was not provided.) Depends on D95224222 Differential Revision: D95837435 Co-authored-by: Dark Knight <generatedunixname89002005232357@meta.com>
1 parent e3b0b33 commit f09bd55

1 file changed

Lines changed: 1 addition & 16 deletions

File tree

examples/models/llama/runner/targets.bzl

Lines changed: 1 addition & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -6,21 +6,6 @@ def _get_operator_lib(aten = False):
66
else:
77
return ["//executorch/configurations:optimized_native_cpu_ops", "//executorch/extension/llm/custom_ops:custom_ops"]
88

9-
def _get_torchao_lowbit_deps():
10-
"""Returns torchao lowbit kernel deps for shared embedding and linear on ARM builds."""
11-
if runtime.is_oss:
12-
return []
13-
else:
14-
# Use select to conditionally include torchao lowbit kernels only on ARM64 builds
15-
# These kernels are only available for aarch64 architecture
16-
return select({
17-
"DEFAULT": [],
18-
"ovr_config//cpu:arm64": [
19-
"//xplat/pytorch/ao/torchao/csrc/cpu/shared_kernels/embedding_xbit:op_embedding_xbit_executorch",
20-
"//xplat/pytorch/ao/torchao/csrc/cpu/shared_kernels/linear_8bit_act_xbit_weight:op_linear_8bit_act_xbit_weight_executorch",
21-
],
22-
})
23-
249
def get_qnn_dependency():
2510
# buck build -c executorch.enable_qnn=true //executorch/examples/models/llama/runner:runner
2611
# Check if QNN is enabled before including the dependency
@@ -63,7 +48,7 @@ def define_common_targets():
6348
"//pytorch/tokenizers:llama2c_tokenizer",
6449
"//pytorch/tokenizers:hf_tokenizer",
6550
"//pytorch/tokenizers:regex_lookahead",
66-
] + (_get_operator_lib(aten)) + _get_torchao_lowbit_deps() + ([
51+
] + (_get_operator_lib(aten)) + ([
6752
# Vulkan API currently cannot build on some platforms (e.g. Apple, FBCODE)
6853
# Therefore enable it explicitly for now to avoid failing tests
6954
"//executorch/backends/vulkan:vulkan_backend_lib",

0 commit comments

Comments
 (0)