fix ipex stats bug (#1555)

Signed-off-by: xin3he <xin3.he@intel.com>
intel · Jan 22, 2024 · 5af3834 · 5af3834
1 parent 1d60f61
commit 5af3834
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
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