Skip to content

Commit baa2968

Browse files
committed
Testing fix to relative imports error
1 parent f4559d8 commit baa2968

File tree

1 file changed

+16
-0
lines changed

1 file changed

+16
-0
lines changed

codegen/tools/gen_oplist.py

Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -15,10 +15,26 @@
1515

1616
try:
1717
from executorch.codegen.parse import strip_et_fields
18+
from executorch.exir._serialize import _deserialize_pte_binary
19+
from executorch.exir.schema import (
20+
EValue,
21+
KernelCall,
22+
OptionalTensorList,
23+
Tensor,
24+
TensorList
25+
)
1826
except ImportError:
1927
# If we build from source, executorch.codegen is not available.
2028
# We can use relative import instead.
2129
from ..parse import strip_et_fields
30+
from ..exir._serialize import _deserialize_pte_binary
31+
from ..exir.schema import (
32+
EValue,
33+
KernelCall,
34+
OptionalTensorList,
35+
Tensor,
36+
TensorList
37+
)
2238

2339
from torchgen.gen import LineLoader, parse_native_yaml_struct
2440
from torchgen.selective_build.operator import SelectiveBuildOperator

0 commit comments

Comments
 (0)