@@ -609,17 +609,17 @@ def query_fw_capability(self, model):
609
609
for precision in precisions :
610
610
if precision == 'fp16' and self .device == 'cpu' and os .getenv ('FORCE_FP16' ) != '1' :
611
611
continue
612
- if precision in self . query_handler .get_quantization_capability ():
613
- special_config_types = list (self . query_handler .get_quantization_capability () \
612
+ if precision in query .get_quantization_capability ():
613
+ special_config_types = list (query .get_quantization_capability () \
614
614
[precision ].keys ())
615
- default_config = self . query_handler .get_quantization_capability () \
615
+ default_config = query .get_quantization_capability () \
616
616
[precision ]['default' ]
617
617
else :
618
618
special_config_types = {}
619
619
default_config = {'weight' : {'dtype' : precision },
620
620
'activation' : {'dtype' : precision }}
621
- optypes = self . query_handler .get_op_types_by_precision (precision ) if \
622
- self . query_handler .get_op_types_by_precision (precision ) != ['*' ] else \
621
+ optypes = query .get_op_types_by_precision (precision ) if \
622
+ query .get_op_types_by_precision (precision ) != ['*' ] else \
623
623
optype_wise .keys ()
624
624
for op in optypes :
625
625
if op not in quantizable_optype :
@@ -628,7 +628,7 @@ def query_fw_capability(self, model):
628
628
op_capability = copy .deepcopy (default_config )
629
629
else :
630
630
op_capability = copy .deepcopy (
631
- self . query_handler .get_quantization_capability ()[precision ][op ])
631
+ query .get_quantization_capability ()[precision ][op ])
632
632
633
633
if precision in ['int8' , 'uint8' ]:
634
634
if self .static :
0 commit comments