Add MatMul, RQMatMul, MatMulInteger

Author: lukamac

Deeploy/OperatorDescriptor.py

@@ -466,6 +466,23 @@ def canonicalize(self, node: gs.Node, opset: int) -> bool:
     attrDescriptors = [],
 )
 
+matMulDesc = OperatorDescriptor(
+    inputDescriptor = IoDesc(["A", "B"]),
+    outputDescriptor = IoDesc("data_out"),
+    attrDescriptors = [],
+)
+
+rqMatMulDesc = RequantizedOperatorDescriptor(
+    inputDescriptor = IoDesc(["A", "B", "add", "mul"]),
+    outputDescriptor = IoDesc("data_out"),
+    attrDescriptors = [
+        # RequantizedShift attrs
+        AttrDesc("n_levels", IntUnpack),
+        AttrDesc("signed", BoolUnpack),
+        AttrDesc("div", IntUnpack),
+    ],
+)
+
 defaultOperatorDescriptors: Dict[str, OperatorDescriptor] = {
     "Add": addDesc,
     "Concat": concatDesc,
@@ -480,11 +497,14 @@ def canonicalize(self, node: gs.Node, opset: int) -> bool:
     "ITAPartialMax": itaPartialMaxDesc,
     "IntegerDiv": integerDivDescriptor,
     "LayerNormalization": layerNormalizationDesc,
+    "MatMul": matMulDesc,
+    "MatMulInteger": matMulDesc,
     "MaxPool": maxPoolDesc,
     "Mul": mulDesc,
     "Pad": padDescOld,
     "Quant": quantDesc,
     "RQIntegerDiv": requantizedIntegerDivDescriptor,
+    "RQMatMul": rqMatMulDesc,
     "ReduceMean": reduceMeanDesc,
     "ReduceSum": reduceSumDesc,
     "RequantizedConv": requantizedConvDesc,