Add MLIR support (#1044)

source/mlir-metadata.json

@@ -1878,7 +1878,8 @@
     "attributes": [
       { "name": "reduce_dimension", "type": "I64Attr" }
     ],
-    "assemblyFormat": "$operand `,` `dim` `=` $reduce_dimension attr-dict `:` functional-type(operands, results)"
+    "assemblyFormat": "$operand `,` `dim` `=` $reduce_dimension attr-dict `:` functional-type(operands, results)",
+    "category": "Activation"
   },
   {
     "name": "asuka.specify_type",
@@ -1908,7 +1909,8 @@
     "attributes": [
       { "name": "split_dimension", "type": "I64Attr" }
     ],
-    "assemblyFormat": "$operand `,` `dim` `=` $split_dimension attr-dict `:` functional-type(operands, results)"
+    "assemblyFormat": "$operand `,` `dim` `=` $split_dimension attr-dict `:` functional-type(operands, results)",
+    "category": "Tensor"
   },
   {
     "name": "asuka.sub",
@@ -1932,7 +1934,8 @@
     "outputs": [
       { "name": "result", "type": "AS_AnyTensor" }
     ],
-    "assemblyFormat": "$operand attr-dict `:` functional-type(operands, results)"
+    "assemblyFormat": "$operand attr-dict `:` functional-type(operands, results)",
+    "category": "Activation"
   },
   {
     "name": "asuka.transpose",
@@ -1945,7 +1948,8 @@
     "attributes": [
       { "name": "dims", "type": "DenseI64ArrayAttr" }
     ],
-    "assemblyFormat": "$operand `,` `dims` `=` $dims attr-dict `:` functional-type(operands, results)"
+    "assemblyFormat": "$operand `,` `dims` `=` $dims attr-dict `:` functional-type(operands, results)",
+    "category": "Transform"
   },
   {
     "name": "asuka.trilu",
@@ -38545,7 +38549,8 @@
     ],
     "attributes": [
       { "name": "num_splits", "type": "ConfinedAttr" }
-    ]
+    ],
+    "category": "Tensor"
   },
   {
     "name": "tfl.split_v",

tools/mlir_script.js

@@ -361,12 +361,12 @@ const main = async () => {
         }
         if (Object.keys(operation).length > 1) {
             if (!operation.category) {
-                const name = operation.name.replace(/^(stablehlo|chlo|affine|linalg|memref|quant|vector|tosa|tfl|tf|onnx|torch\.aten|gpu)\./, '');
+                const name = operation.name.replace(/^(asuka|stablehlo|chlo|affine|linalg|memref|quant|vector|tosa|tfl|tf|onnx|torch\.aten|gpu)\./, '');
                 if (['reshape', 'broadcast_in_dim', 'dynamic_reshape', 'Reshape', 'Shape', 'Size', 'ConstantOfShape'].indexOf(name) !== -1) {
                     operation.category = 'Shape';
                 } else if (['transpose', 'reverse', 'pad', 'Transpose', 'Pad'].indexOf(name) !== -1) {
                     operation.category = 'Transform';
-                } else if (['slice', 'dynamic_slice', 'gather', 'scatter', 'Slice', 'Gather', 'Scatter', 'concatenate'].indexOf(name) !== -1) {
+                } else if (['slice', 'split', 'dynamic_slice', 'gather', 'scatter', 'Slice', 'Gather', 'Scatter', 'concatenate'].indexOf(name) !== -1) {
                     operation.category = 'Tensor';
                 } else if (['tanh', 'Sigmoid', 'Tanh', 'Relu', 'Softmax', 'softmax', 'sigmoid', 'relu'].indexOf(name) !== -1) {
                     operation.category = 'Activation';