@@ -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