Support torch > 1.8.0

Signed-off-by: Cheng, Penghui <[email protected]>
intel · Aug 14, 2023 · b40c0b1 · b40c0b1
1 parent 6b92ebb
commit b40c0b1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/neural_compressor/adaptor/pytorch.py b/neural_compressor/adaptor/pytorch.py
@@ -847,7 +847,7 @@ def __init__(self, framework_specific_info):
         self.fp32_results = []
         self.fp32_preds_as_label = False
 
-        if self.version.release >= Version("2.0").release:
+        if self.version.release >= Version("1.8").release:
             static_quant_mapping = tq.quantization_mappings.get_default_static_quant_module_mappings()
             self.fused_op_list = \
                 [static_quant_mapping[key] for key in static_quant_mapping if "intrinsic." in str(key)]