Skip to content

Commit d7113f8

Browse files
committed
Fix MHSA requant information being unpacked to Int unstead of IntTuple
1 parent 378594e commit d7113f8

File tree

1 file changed

+33
-33
lines changed

1 file changed

+33
-33
lines changed

Deeploy/OperatorDescriptor.py

Lines changed: 33 additions & 33 deletions
Original file line numberDiff line numberDiff line change
@@ -527,26 +527,26 @@ def canonicalize(self, node: gs.Node, opset: int) -> bool:
527527
["q", "k", "v", "wq_weight", "wq_bias", "wk_weight", "wk_bias", "wv_weight", "wv_bias", "wo_weight", "wo_bias"]),
528528
outputDescriptor = IoDesc("data_out"),
529529
attrDescriptors = [
530-
AttrDesc("preattn_requant_mul", IntUnpack),
531-
AttrDesc("preattn_requant_div", IntUnpack),
532-
AttrDesc("normalizer_requant_mul", IntUnpack),
533-
AttrDesc("normalizer_requant_shift", IntUnpack),
534-
AttrDesc("normalizer_requant_div", IntUnpack),
535-
AttrDesc("postattn_requant_mul", IntUnpack),
536-
AttrDesc("postattn_requant_shift", IntUnpack),
537-
AttrDesc("postattn_requant_div", IntUnpack),
538-
AttrDesc("wo_requant_mul", IntUnpack),
539-
AttrDesc("wo_requant_shift", IntUnpack),
540-
AttrDesc("wo_requant_div", IntUnpack),
541-
AttrDesc("wq_requant_mul", IntUnpack),
542-
AttrDesc("wq_requant_shift", IntUnpack),
543-
AttrDesc("wq_requant_div", IntUnpack),
544-
AttrDesc("wk_requant_mul", IntUnpack),
545-
AttrDesc("wk_requant_shift", IntUnpack),
546-
AttrDesc("wk_requant_div", IntUnpack),
547-
AttrDesc("wv_requant_mul", IntUnpack),
548-
AttrDesc("wv_requant_shift", IntUnpack),
549-
AttrDesc("wv_requant_div", IntUnpack),
530+
AttrDesc("preattn_requant_mul", IntTupleUnpack),
531+
AttrDesc("preattn_requant_div", IntTupleUnpack),
532+
AttrDesc("normalizer_requant_mul", IntTupleUnpack),
533+
AttrDesc("normalizer_requant_shift", IntTupleUnpack),
534+
AttrDesc("normalizer_requant_div", IntTupleUnpack),
535+
AttrDesc("postattn_requant_mul", IntTupleUnpack),
536+
AttrDesc("postattn_requant_shift", IntTupleUnpack),
537+
AttrDesc("postattn_requant_div", IntTupleUnpack),
538+
AttrDesc("wo_requant_mul", IntTupleUnpack),
539+
AttrDesc("wo_requant_shift", IntTupleUnpack),
540+
AttrDesc("wo_requant_div", IntTupleUnpack),
541+
AttrDesc("wq_requant_mul", IntTupleUnpack),
542+
AttrDesc("wq_requant_shift", IntTupleUnpack),
543+
AttrDesc("wq_requant_div", IntTupleUnpack),
544+
AttrDesc("wk_requant_mul", IntTupleUnpack),
545+
AttrDesc("wk_requant_shift", IntTupleUnpack),
546+
AttrDesc("wk_requant_div", IntTupleUnpack),
547+
AttrDesc("wv_requant_mul", IntTupleUnpack),
548+
AttrDesc("wv_requant_shift", IntTupleUnpack),
549+
AttrDesc("wv_requant_div", IntTupleUnpack),
550550
AttrDesc("Delta", IntUnpack),
551551
AttrDesc("eps", IntUnpack),
552552
AttrDesc("act_type", IntUnpack),
@@ -584,23 +584,23 @@ def canonicalize(self, node: gs.Node, opset: int) -> bool:
584584
"wo_bias"]),
585585
outputDescriptor = IoDesc("data_out"),
586586
attrDescriptors = [
587-
AttrDesc("preattn_requant_mul", IntUnpack),
588-
AttrDesc("preattn_requant_div", IntUnpack),
589-
AttrDesc("postattn_requant_mul", IntUnpack),
590-
AttrDesc("postattn_requant_div", IntUnpack),
591-
AttrDesc("wo_requant_mul", IntUnpack),
592-
AttrDesc("wo_requant_div", IntUnpack),
593-
AttrDesc("wq_requant_mul", IntUnpack),
594-
AttrDesc("wq_requant_div", IntUnpack),
595-
AttrDesc("wk_requant_mul", IntUnpack),
596-
AttrDesc("wk_requant_div", IntUnpack),
597-
AttrDesc("wv_requant_mul", IntUnpack),
598-
AttrDesc("wv_requant_div", IntUnpack),
587+
AttrDesc("preattn_requant_mul", IntTupleUnpack),
588+
AttrDesc("preattn_requant_div", IntTupleUnpack),
589+
AttrDesc("postattn_requant_mul", IntTupleUnpack),
590+
AttrDesc("postattn_requant_div", IntTupleUnpack),
591+
AttrDesc("wo_requant_mul", IntTupleUnpack),
592+
AttrDesc("wo_requant_div", IntTupleUnpack),
593+
AttrDesc("wq_requant_mul", IntTupleUnpack),
594+
AttrDesc("wq_requant_div", IntTupleUnpack),
595+
AttrDesc("wk_requant_mul", IntTupleUnpack),
596+
AttrDesc("wk_requant_div", IntTupleUnpack),
597+
AttrDesc("wv_requant_mul", IntTupleUnpack),
598+
AttrDesc("wv_requant_div", IntTupleUnpack),
599599
AttrDesc("n_levels", IntUnpack),
600600
AttrDesc("dim", IntUnpack),
601601
AttrDesc("dim_head", IntUnpack),
602602
AttrDesc("heads", IntUnpack),
603-
AttrDesc("signed", IntUnpack),
603+
AttrDesc("signed", BoolUnpack),
604604
],
605605
)
606606

0 commit comments

Comments
 (0)