Skip to content

Commit fa17427

Browse files
authored
Fix pt2e test_qat_preserve_source_fn_stack (#3149)
**Summary:** A recent change in `torch.export` now inlines the submodules in the `source_fn_stack`. We should adjust our test accordingly to unbreak CI. **Test Plan:** python test/quantization/pt2e/test_quantize_pt2e_qat.py -k test_qat_preserve_source_fn_stack
1 parent 29f6758 commit fa17427

File tree

1 file changed

+0
-6
lines changed

1 file changed

+0
-6
lines changed

test/quantization/pt2e/test_quantize_pt2e_qat.py

Lines changed: 0 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -686,12 +686,6 @@ def get_source_fn(node: torch.fx.Node):
686686
self.assertNotEqual(get_source_fn(second_conv), get_source_fn(second_relu))
687687
self.assertNotEqual(get_source_fn(first_relu), get_source_fn(second_relu))
688688

689-
# Assert that "backbone" exists only in the second set of conv and relu's partition
690-
self.assertTrue("backbone" not in get_source_fn(first_conv))
691-
self.assertTrue("backbone" not in get_source_fn(first_relu))
692-
self.assertTrue("backbone" in get_source_fn(second_conv))
693-
self.assertTrue("backbone" in get_source_fn(second_relu))
694-
695689
def test_qat_conv_bn_bias_derived_qspec(self):
696690
m = self._get_conv_bn_model()
697691
example_inputs = self.example_inputs

0 commit comments

Comments
 (0)