Skip to content

Commit 3c18467

Browse files
committed
Add missing Gemm and RQGemm descriptors to defaultOperatorDescriptors
1 parent 5d634a4 commit 3c18467

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

Deeploy/OperatorDescriptor.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -717,6 +717,7 @@ def canonicalize(self, node: gs.Node, opset: int) -> bool:
717717
"Flatten": flattenDesc,
718718
"Gather": gatherDesc,
719719
"Gelu": geluDesc,
720+
"Gemm": gemmDesc,
720721
"ITAMax": itaMaxDesc,
721722
"ITAPartialMax": itaPartialMaxDesc,
722723
"IntegerDiv": integerDivDescriptor,
@@ -730,6 +731,7 @@ def canonicalize(self, node: gs.Node, opset: int) -> bool:
730731
"Mul": mulDesc,
731732
"Pad": padDescOld,
732733
"Quant": quantDesc,
734+
"RQGemm": rqGemmDesc,
733735
"RQIntegerDiv": requantizedIntegerDivDescriptor,
734736
"RQMatMul": rqMatMulDesc,
735737
"ReduceMean": reduceMeanDesc,

0 commit comments

Comments
 (0)