@@ -722,40 +722,34 @@ def convert_prelu_post_op(post_op):
722
722
return benchdnn_postops
723
723
724
724
725
- def convert_scales ( scales , prim_kind ):
725
+ def convert_quantization ( q_param , prim_kind , def_value , def_type ):
726
726
res = []
727
- for arg in scales .keys ():
728
- s = scales [arg ]
729
- policy = convert_scale_policy (s ["mask" ], prim_kind )
730
- benchdnn_scale = arg + ":" + policy
727
+ for arg in q_param .keys ():
728
+ p = q_param [arg ]
729
+ policy = convert_scale_policy (p ["mask" ], prim_kind )
730
+ benchdnn_p = arg + ":" + policy
731
731
if policy == "common" :
732
- benchdnn_scale += ":0.5"
733
- dt = s ["data_type" ]
734
- groups = s ["groups" ]
735
- if dt != "f32" or groups != "" :
736
- benchdnn_scale += ":" + dt
732
+ benchdnn_p += ":" + def_value
733
+ dt = p ["data_type" ]
734
+ groups = p ["groups" ]
735
+ if dt != def_type or groups != "" :
736
+ benchdnn_p += ":" + dt
737
737
if groups != "" :
738
- benchdnn_scale += ":" + groups
739
- res .append (benchdnn_scale )
738
+ benchdnn_p += ":" + groups
739
+ res .append (benchdnn_p )
740
740
return "+" .join (res )
741
741
742
742
743
+ def convert_scales (scales , prim_kind ):
744
+ return convert_quantization (
745
+ q_param = scales , prim_kind = prim_kind , def_value = "0.5" , def_type = "f32"
746
+ )
747
+
748
+
743
749
def convert_zero_points (zero_points , prim_kind ):
744
- res = []
745
- for arg in zero_points .keys ():
746
- zp = zero_points [arg ]
747
- policy = convert_zp_policy (zp ["mask" ], prim_kind )
748
- benchdnn_zp = arg + ":" + policy
749
- if policy == "common" :
750
- benchdnn_zp += ":1"
751
- dt = zp ["data_type" ]
752
- groups = zp ["groups" ]
753
- if dt != "s32" or groups != "" :
754
- benchdnn_zp += ":" + dt
755
- if groups != "" :
756
- benchdnn_zp += ":" + groups
757
- res .append (benchdnn_zp )
758
- return "+" .join (res )
750
+ return convert_quantization (
751
+ q_param = zero_points , prim_kind = prim_kind , def_value = "1" , def_type = "s32"
752
+ )
759
753
760
754
761
755
def convert_scratchpad_mode (scratchpad_mode , prim_kind ):
0 commit comments