fix ipex stats bug (#1555)

xin3he · web-flow · commit 5af383465190 · 2024-01-22T15:47:27.000+08:00
Signed-off-by: xin3he &lt;xin3.he@intel.com&gt;
diff --git a/neural_compressor/adaptor/pytorch.py b/neural_compressor/adaptor/pytorch.py
@@ -2599,7 +2599,8 @@ def get_non_quant_modules(self, model_kwargs):
     "<class 'torch.nn.modules.conv.Conv2d'>": "Conv2d",
     "<class 'torch.nn.modules.conv.Conv3d'>": "Conv3d",
     "<class 'torch.nn.modules.activation.ReLU'>": "ReLU",
-    "<method 'add' of 'torch._C._TensorBase' objects>": "add",
+    "<method 'add' of 'torch._C._TensorBase' objects>": "add",  # for IPEX < 2.2
+    "<method 'add' of 'torch._C.TensorBase' objects>": "add",  # for IPEX >= 2.2
     "<class 'torch.nn.modules.pooling.AdaptiveAvgPool2d'>": "AdaptiveAvgPool2d",
     "Linear_Relu": "Linear",
     "<class 'torch.nn.modules.linear.Linear'>": "Linear",
diff --git a/test/algorithm/test_smooth_quant.py b/test/algorithm/test_smooth_quant.py
@@ -41,6 +41,7 @@ def __init__(self):
     def forward(self, x):
         out = self.fc1(x)
         out = self.fc2(out)
+        out = out + out
         return out
 
 
