[quant][pt2] Add test for inplace add (#102867)

andrewor14 · pytorchmergebot · commit 2e8d2a2e6947 · 2023-06-07T19:43:28.000Z
Summary: This was broken after the recent partitioner refactors. Test Plan: python test/test_quantization.py TestQuantizePT2E.test_qat_inplace_add_relu Differential Revision: D46402378 Pull Request resolved: #102867 Approved by: https://github.com/jerryzh168
diff --git a/test/quantization/pt2e/test_quantize_pt2e.py b/test/quantization/pt2e/test_quantize_pt2e.py
@@ -1208,6 +1208,28 @@ def test_prepare_qat_conv_bn_relu_fusion(self):
             m1, example_inputs, is_per_channel=True, has_relu=True
         )
 
+    def test_qat_inplace_add_relu(self):
+        class M(torch.nn.Module):
+            def __init__(self):
+                super().__init__()
+                self.conv = torch.nn.Conv2d(1, 1, 1)
+                self.relu = torch.nn.ReLU(inplace=True)
+
+            def forward(self, x):
+                x0 = x
+                x = self.conv(x)
+                x += x0
+                x = self.relu(x)
+                return x
+
+        example_inputs = (torch.randn(1, 1, 3, 3),)
+        self._verify_symmetric_qnnpack_qat_numerics(
+            M(), example_inputs, is_per_channel=False, verify_convert=True,
+        )
+        self._verify_symmetric_qnnpack_qat_numerics(
+            M(), example_inputs, is_per_channel=True, verify_convert=True,
+        )
+
     def test_prepare_qat_conv_bn_fusion_getitem_placeholder(self):
         """
         Test this special case seen in resnet18:
