test: add test for Gemm with integers

TaceoLabs · Feb 16, 2024 · 89de344 · 89de344
1 parent 2bffd54
commit 89de344
Show file tree

Hide file tree

Showing 3 changed files with 56 additions and 0 deletions.
diff --git a/mlir-assigner/tests/Ops/Onnx/Gemm/GemmInt.json b/mlir-assigner/tests/Ops/Onnx/Gemm/GemmInt.json
@@ -0,0 +1,53 @@
+[
+  {
+    "memref": {
+      "idx": 0,
+      "data": [
+        -7301, -32382, -21319, 12369, -6339, 26890, 12385, -21692, 25678,
+        -21372, 7721, 28175, 32735, -2135, -15232, 12237
+      ],
+      "dims": [1, 16],
+      "type": "int"
+    }
+  },
+  {
+    "memref": {
+      "idx": 1,
+      "data": [
+        -9888, 238, -14497, 11426, 21358, -6714, 8900, 31279, 25539, 18251,
+        -233, 24789, -31649, 6890, -25061, 3655, 8255, 16388, 4604, -3083,
+        -14374, -22534, -23088, 6545, -14257, -21374, -17442, -7534, -17748,
+        -23683, -2633, -10258, 5629, -26725, 32707, -13939, 840, 30317, -23721,
+        -11844, -17924, 24910, -18968, -27365, -31857, -8749, 833, -17245, 9830,
+        -29882, 27940, -14856, 16936, 16871, 20919, 8556, 5394, 1523, -25147,
+        27289, -25636, -25937, 15477, 14239, 27245, -7941, -10435, -6071,
+        -30011, 394, -12021, -3368, 13146, 26424, -152, 5961, 2412, -12226,
+        31539, -22306, 31246, -27137, -20442, 25959, -11895, -26850, -4408,
+        11092, 26272, -7548, 30486, -10409, 7576, 30402, 32117, -12078, 4149,
+        9385, -13996, 23657, -7724, 11043, -14920, 20336, 22599, -10035, -7875,
+        -18044, 28875, -26843, -26231, 7855, 15947, 29190, -4278, -25381,
+        -16350, 19469, -25831, -30015, 4995, 866, -17414, 14269, 4252, -9091,
+        18425, -26417, 6826, 10118, 18499, 27766, -1313, 1756, -17977, -16550,
+        27152, 13994, 23810, 26776, -5106, -17564, 4905, -6326, 23489, 21688,
+        -21159, -28445, -886, 29171, -22092, 17380, -3481, -26942, 7526, 14023,
+        -19036, 20024, 19528, 19179, 16043, -21341, 30662, -24813, 18735, 18428,
+        -18069, -29081, -689, -9969, -3201, 13793, 11778, 11781, 12195, 4677,
+        -14815, -1633, 32545, -2545, -26616, 11382, -7665, -18427, 735, 29520,
+        -81, 5556, -22666, -1231, -16522, -25874
+      ],
+      "dims": [12, 16],
+      "type": "int"
+    }
+  },
+  {
+    "memref": {
+      "idx": 2,
+      "data": [
+        1255, 11823, -18085, -24954, -17146, 21451, 14861, 29958, 3558, -6807,
+        21488, -27584
+      ],
+      "dims": [12],
+      "type": "int"
+    }
+  }
+]
diff --git a/mlir-assigner/tests/Ops/Onnx/Gemm/GemmInt.onnx b/mlir-assigner/tests/Ops/Onnx/Gemm/GemmInt.onnx
diff --git a/mlir-assigner/tests/Ops/Onnx/Gemm/GemmInt.res b/mlir-assigner/tests/Ops/Onnx/Gemm/GemmInt.res
@@ -0,0 +1,3 @@
+Result:
+memref<1x12xi64>[-31252652, -2542027060, -2430328551, 371570543, -378097300, 789864332, 1700585817, -394018956, 818092837, -424166328, 1188315658, -1045134730]
+232