fixed some bugs

Signed-off-by: yiliu30 <[email protected]>

Author: yiliu30

neural_compressor/adaptor/torch_utils/util.py

@@ -609,7 +609,10 @@ def output_hook(self, input, output):
         from torch.quantization.quantize_fx import prepare_fx,convert_fx
         # do quantization
         if adaptor.sub_module_list is None:
-            tmp_model = prepare_fx(tmp_model, fx_op_cfgs,)
+            if _torch_version_greater_than("1.13.0"):
+                tmp_model = prepare_fx(tmp_model, fx_op_cfgs, example_inputs=None)
+            else:
+                tmp_model = prepare_fx(tmp_model, fx_op_cfgs,)
         else:
             PyTorch_FXAdaptor.prepare_sub_graph(adaptor.sub_module_list, fx_op_cfgs, \
                                                 tmp_model, prefix='')
@@ -658,7 +661,10 @@ def output_hook(self, input, output):
         from torch.quantization.quantize_fx import prepare_fx,convert_fx
         # do quantization
         if adaptor.sub_module_list is None:
-            tmp_model = prepare_fx(tmp_model, fx_op_cfgs,)
+            if _torch_version_greater_than("1.13.0"):
+                tmp_model = prepare_fx(tmp_model, fx_op_cfgs, example_inputs=None)
+            else:
+                tmp_model = prepare_fx(tmp_model, fx_op_cfgs,)
         else:
             PyTorch_FXAdaptor.prepare_sub_graph(adaptor.sub_module_list, fx_op_cfgs, \
                                                 tmp_model, prefix='')
@@ -722,7 +728,10 @@ def output_hook(self, input, output):
             from torch.quantization.quantize_fx import prepare_fx,convert_fx
             # do quantization
             if adaptor.sub_module_list is None:
-                tmp_model = prepare_fx(tmp_model, fx_op_cfgs,)
+                if _torch_version_greater_than("1.13.0"):
+                    tmp_model = prepare_fx(tmp_model, fx_op_cfgs,example_inputs=None)
+                else:
+                    tmp_model = prepare_fx(tmp_model, fx_op_cfgs,)
             else:
                 PyTorch_FXAdaptor.prepare_sub_graph(adaptor.sub_module_list, fx_op_cfgs, \
                                                     tmp_model, prefix='')
@@ -750,3 +759,17 @@ def output_hook(self, input, output):
     ordered_ops = sorted(fallback_order.keys(), key=lambda key: fallback_order[key], \
                                             reverse=False)
     return ordered_ops
+
+def get_torch_version():
+    from packaging.version import Version
+    try:
+        torch_version = torch.__version__.split('+')[0]
+    except ValueError as e:  # pragma: no cover
+        assert False, 'Got an unknown version of torch: {}'.format(e)
+    version = Version(torch_version)
+    return version
+
+def _torch_version_greater_than(version_number : str):
+    from packaging.version import Version
+    torch_version = get_torch_version()
+    return torch_version.release >= Version(version_number).release  # pragma: no cover