Skip to content

Commit 248fb2e

Browse files
committed
scripts: verbose_converter: benchdnn: reuse conversion for sc and zp
1 parent 6fc2b7f commit 248fb2e

File tree

1 file changed

+21
-27
lines changed

1 file changed

+21
-27
lines changed

scripts/verbose_converter/src/benchdnn_generator.py

+21-27
Original file line numberDiff line numberDiff line change
@@ -722,40 +722,34 @@ def convert_prelu_post_op(post_op):
722722
return benchdnn_postops
723723

724724

725-
def convert_scales(scales, prim_kind):
725+
def convert_quantization(q_param, prim_kind, def_value, def_type):
726726
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
731731
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
737737
if groups != "":
738-
benchdnn_scale += ":" + groups
739-
res.append(benchdnn_scale)
738+
benchdnn_p += ":" + groups
739+
res.append(benchdnn_p)
740740
return "+".join(res)
741741

742742

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+
743749
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+
)
759753

760754

761755
def convert_scratchpad_mode(scratchpad_mode, prim_kind):

0 commit comments

Comments
 (0)