We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent dfa2d7a commit 44d8d54Copy full SHA for 44d8d54
optimum/exporters/executorch/recipes/xnnpack.py
@@ -17,6 +17,7 @@
17
18
from tabulate import tabulate
19
from torch.export import ExportedProgram
20
+from torchao.utils import unwrap_tensor_subclass
21
22
from executorch.backends.xnnpack.partition.xnnpack_partitioner import XnnpackPartitioner
23
from executorch.devtools.backend_debug import get_delegation_info
@@ -104,6 +105,7 @@ def _lower_to_executorch(
104
105
)
106
return {pte_name: et_prog}
107
108
+ model = unwrap_tensor_subclass(model)
109
exported_progs = model.export()
110
111
if (
0 commit comments