Skip to content

Commit 5417019

Browse files
committed
reset dynamo in for lops
1 parent d03a22b commit 5417019

File tree

2 files changed

+2
-0
lines changed

2 files changed

+2
-0
lines changed

backends/xnnpack/test/ops/test_check_quant_params.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -49,6 +49,7 @@ def inject_invalid_scale_in_per_tensor(aten):
4949
return inject_invalid_scale_in_per_tensor
5050

5151
def _test_check_quant_message(self, ep_modifier, expected_message):
52+
torch._dynamo.reset()
5253
mod = torch.nn.Linear(10, 10)
5354
quantizer = XNNPACKQuantizer()
5455
captured = export_for_training(mod, (torch.randn(1, 10),)).module()

backends/xnnpack/test/passes/test_convert_to_linear.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -24,6 +24,7 @@ def test_fp32_convert_to_linear(self):
2424
bias_vals = [True, True, False]
2525

2626
for i, _ in enumerate(in_sizes):
27+
torch._dynamo.reset()
2728
in_size = int(in_sizes[i])
2829
input_size = int(input_sizes[i])
2930
output_size = int(output_sizes[i])

0 commit comments

Comments
 (0)