Skip to content

Commit 41dd47d

Browse files
larryliu0820Zonglin Peng
authored and
Zonglin Peng
committed
Add optimized kernels to executorch pybindings
Differential Revision: D70540908 Pull Request resolved: #8943
1 parent 6bf4e5b commit 41dd47d

File tree

4 files changed

+6
-5
lines changed

4 files changed

+6
-5
lines changed

exir/backend/test/TARGETS

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -107,7 +107,7 @@ python_unittest(
107107
"test_backends.py",
108108
],
109109
preload_deps = [
110-
"//executorch/kernels/portable:custom_ops_generated_lib",
110+
"//executorch/configurations:optimized_native_cpu_ops",
111111
"//executorch/kernels/quantized:custom_ops_generated_lib",
112112
"//executorch/runtime/executor/test:test_backend_compiler_lib",
113113
],

exir/backend/test/demos/rpc/TARGETS

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -47,7 +47,7 @@ python_unittest(
4747
"test_rpc.py",
4848
],
4949
preload_deps = [
50-
"//executorch/kernels/portable:custom_ops_generated_lib",
50+
"//executorch/configurations:optimized_native_cpu_ops",
5151
"//executorch/kernels/quantized:custom_ops_generated_lib",
5252
# the executor backend is prebuilt and linked when building the unit test binary. When it's linked, it'll register the backend.
5353
# It can also be loaded in PyThon runtime via torch.ops.load_library("//executorch/exir/backend/test/demos/rpc:executor_backend")

exir/backend/test/demos/rpc/targets.bzl

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -23,9 +23,9 @@ def define_common_targets():
2323
],
2424
platforms = [ANDROID, CXX],
2525
deps = [
26-
"//executorch/runtime/executor:program",
27-
"//executorch/kernels/portable:generated_lib",
26+
"//executorch/configurations:optimized_native_cpu_ops",
2827
"//executorch/runtime/backend:interface",
28+
"//executorch/runtime/executor:program",
2929
"//executorch/extension/data_loader:buffer_data_loader",
3030
] + MODELS_ATEN_OPS_LEAN_MODE_GENERATED_LIB,
3131
exported_deps = [

shim_et/xplat/executorch/extension/pybindings/pybindings.bzl

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@ load("@fbsource//xplat/executorch/build:runtime_wrapper.bzl", "runtime")
33

44
# Aten ops with portable kernel
55
MODELS_ATEN_OPS_LEAN_MODE_GENERATED_LIB = [
6-
"//executorch/kernels/portable:generated_lib",
6+
"//executorch/configurations:optimized_native_cpu_ops",
77
"//executorch/kernels/quantized:generated_lib",
88
]
99

@@ -41,6 +41,7 @@ MODELS_ATEN_OPS_ATEN_MODE_GENERATED_LIB = [
4141

4242
def executorch_pybindings(python_module_name, srcs = [], cppdeps = [], visibility = ["//executorch/..."], types = [], compiler_flags = []):
4343
runtime.cxx_python_extension(
44+
# @autodeps-skip
4445
name = python_module_name,
4546
srcs = [
4647
"//executorch/extension/pybindings:pybindings.cpp",

0 commit comments

Comments
 (0)