Skip to content

Commit b64466a

Browse files
mcr229facebook-github-bot
authored andcommitted
Make Default Testing Config Canonical Config
Summary: For Tester and our long term testing, we should be using the canonical capture config, so we are testing the right way Differential Revision: D49214927 fbshipit-source-id: 6f726295f1fbb74df4435d6756976271baa25fd2
1 parent 70357ff commit b64466a

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

backends/xnnpack/test/tester/tester.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -124,7 +124,9 @@ def graph_module(self) -> str:
124124
@register_stage
125125
class Export(Stage):
126126
def __init__(self, capture_config: Optional[CaptureConfig] = None):
127-
self.capture_conf = capture_config or get_xnnpack_capture_config()
127+
self.capture_conf = capture_config or get_xnnpack_capture_config(
128+
enable_aot=True, unlift=False
129+
)
128130
self.exir_exported_program = None
129131

130132
def run(self, artifact: torch.nn.Module, inputs) -> None:

0 commit comments

Comments
 (0)