Fixed use_bf16 default value error

Signed-off-by: Cheng, Penghui <[email protected]>

Author: PenghuiCheng

neural_compressor/conf/config.py

@@ -840,7 +840,7 @@ def percent_to_float(data):
             str: ops_schema
         },
     },
-    Optional('use_bf16', default=False): bool,
+    Optional('use_bf16', default=True): bool,
     Optional('graph_optimization'): graph_optimization_schema,
     Optional('mixed_precision'): mixed_precision_schema,
 

neural_compressor/conf/pythonic_config.py

@@ -39,11 +39,12 @@ def __init__(self,
                  max_trials=100,
                  performance_only=False,
                  reduce_range=None,
-                 use_bf16=False,
+                 use_bf16=True,
                  accuracy_criterion=accuracy_criterion):
+        extra_precisions = ["bf16"] if use_bf16 else []
         super().__init__(inputs, outputs, backend, device, calibration_sampling_size, op_type_list,
                          op_name_list, strategy, objective, timeout, max_trials, performance_only,
-                         reduce_range, use_bf16, accuracy_criterion)
+                         reduce_range, extra_precisions, accuracy_criterion)
         self._approach = approach
 
     @property

neural_compressor/config.py

@@ -530,16 +530,16 @@ def strategy(self, strategy):
 
 class PostTrainingQuantConfig(_BaseQuantizationConfig):
     def __init__(self,
-                 device='cpu',
+                 device="cpu",
                  backend="NA",
                  inputs=[],
                  outputs=[],
-                 approach='auto',
+                 approach="auto",
                  calibration_sampling_size=[100],
                  op_type_list=None,
                  op_name_list=None,
                  reduce_range=None,
-                 extra_precisions = [],
+                 extra_precisions = ["bf16"],
                  tuning_criterion=tuning_criterion,
                  accuracy_criterion=accuracy_criterion,
     ):
@@ -578,7 +578,7 @@ def __init__(self,
                  op_type_list=None,
                  op_name_list=None,
                  reduce_range=None,
-                 extra_precisions=[]):
+                 extra_precisions=["bf16"]):
         super().__init__(inputs=inputs, outputs=outputs, device=device, backend=backend,
                          op_type_list=op_type_list, op_name_list=op_name_list,
                          reduce_range=reduce_range, extra_precisions=extra_precisions)