Skip to content

Commit 907c030

Browse files
committed
Updating targets.bzl to expose needed schema class defs
1 parent 1871316 commit 907c030

File tree

2 files changed

+14
-2
lines changed

2 files changed

+14
-2
lines changed

codegen/targets.bzl

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -67,6 +67,18 @@ def define_common_targets():
6767
],
6868
)
6969

70+
runtime.python_library(
71+
name = "exir",
72+
srcs = [
73+
"_serialize.py",
74+
"schema.py",
75+
],
76+
base_module = "executorch.exir",
77+
visibility = [
78+
"//executorch/codegen/...",
79+
],
80+
)
81+
7082
runtime.python_binary(
7183
name = "gen",
7284
main_module = "executorch.codegen.gen",

codegen/tools/gen_oplist.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -27,8 +27,8 @@
2727
# If we build from source, executorch.codegen is not available.
2828
# We can use relative import instead.
2929
from ..parse import strip_et_fields
30-
from ...exir._serialize import _deserialize_pte_binary
31-
from ...exir.schema import (
30+
from ..exir._serialize import _deserialize_pte_binary
31+
from ..exir.schema import (
3232
EValue,
3333
KernelCall,
3434
OptionalTensorList,

0 commit comments

Comments
 (0)