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 ef9c057 commit f919020Copy full SHA for f919020
backends/xnnpack/test/tester/tester.py
@@ -124,7 +124,9 @@ def graph_module(self) -> str:
124
@register_stage
125
class Export(Stage):
126
def __init__(self, capture_config: Optional[CaptureConfig] = None):
127
- self.capture_conf = capture_config or get_xnnpack_capture_config()
+ self.capture_conf = capture_config or get_xnnpack_capture_config(
128
+ enable_aot=True, unlift=False
129
+ )
130
self.exir_exported_program = None
131
132
def run(self, artifact: torch.nn.Module, inputs) -> None:
0 commit comments