enable binding

Author: xurui1995

include/gc-c/Dialects.h

@@ -27,6 +27,7 @@ extern "C" {
 #endif
 
 MLIR_DECLARE_CAPI_DIALECT_REGISTRATION(OneDNNGraph, onednn_graph);
+MLIR_DECLARE_CAPI_DIALECT_REGISTRATION(CPURuntime, cpuruntime);
 
 #ifdef __cplusplus
 }

include/gc-c/Passes.h

@@ -24,6 +24,7 @@
 extern "C" {
 #endif
 
+#include "gc/Dialect/CPURuntime/Transforms/CPURuntimePasses.capi.h.inc"
 #include "gc/Transforms/Passes.capi.h.inc"
 #ifdef __cplusplus
 }

include/gc/Dialect/OneDNNGraph/CMakeLists.txt

@@ -1,3 +1,3 @@
 add_mlir_dialect(OneDNNGraphOps onednn_graph)
-# add_mlir_doc(OneDNNGraphOps OneDNNGraphOps gc/Dialect/OneDNNGraph/ -gen-op-doc)
-# add_mlir_doc(OneDNNGraphDialect OneDNNGraphDialect gc/Dialect/OneDNNGraph/ -gen-dialect-doc)
+add_mlir_doc(OneDNNGraphOps OneDNNGraphOps gc/Dialect/OneDNNGraph/ -gen-op-doc)
+add_mlir_doc(OneDNNGraphDialect OneDNNGraphDialect gc/Dialect/OneDNNGraph/ -gen-dialect-doc)

lib/gc/CAPI/CMakeLists.txt

@@ -3,5 +3,7 @@ add_mlir_public_c_api_library(GcCAPI
   Passes.cpp
   LINK_LIBS PUBLIC
   MLIROneDNNGraph
+  MLIRCPURuntimeDialect
   GCPasses
+  MLIRCPURuntimeTransforms
 )

lib/gc/CAPI/Dialects.cpp

@@ -8,7 +8,11 @@
 
 #include "gc-c/Dialects.h"
 #include "gc/Dialect/OneDNNGraph/OneDNNGraphDialect.h"
+#include "gc/Dialect/CPURuntime/IR/CPURuntimeDialect.h"
 #include "mlir/CAPI/Registration.h"
 
 MLIR_DEFINE_CAPI_DIALECT_REGISTRATION(OneDNNGraph, onednn_graph,
-                                      mlir::onednn_graph::OneDNNGraphDialect)
+                                      mlir::onednn_graph::OneDNNGraphDialect)
+
+MLIR_DEFINE_CAPI_DIALECT_REGISTRATION(CPURuntime, cpuruntime,
+                                      mlir::cpuruntime::CPURuntimeDialect)            

lib/gc/CAPI/Passes.cpp

@@ -7,16 +7,20 @@
 //===----------------------------------------------------------------------===//
 
 #include "gc/Transforms/Passes.h"
+#include "gc/Dialect/CPURuntime/Transforms/CPURuntimePasses.h"
 #include "mlir-c/Pass.h"
 #include "mlir/CAPI/Pass.h"
 
+#include "gc/Dialect/CPURuntime/Transforms/CPURuntimePasses.capi.h.inc"
 #include "gc/Transforms/Passes.capi.h.inc"
 using namespace mlir::gc;
+using namespace mlir::cpuruntime;
 
 #ifdef __cplusplus
 extern "C" {
 #endif
 
+#include "gc/Dialect/CPURuntime/Transforms/CPURuntimePasses.capi.cpp.inc"
 #include "gc/Transforms/Passes.capi.cpp.inc"
 
 #ifdef __cplusplus

python/CMakeLists.txt

@@ -52,6 +52,15 @@ declare_mlir_dialect_python_bindings(
     DIALECT_NAME onednn_graph
     )
 
+    declare_mlir_dialect_python_bindings(
+      ADD_TO_PARENT GcPythonSources
+      ROOT_DIR "${CMAKE_CURRENT_SOURCE_DIR}/gc_mlir"
+      TD_FILE dialects/CPURuntimeOps.td
+      SOURCES 
+        dialects/cpuruntime.py
+      DIALECT_NAME cpuruntime
+      )
+
 declare_mlir_python_extension(GcPythonSources.Extension
   MODULE_NAME _gc_mlir
   ADD_TO_PARENT GcPythonSources

python/MainModule.cpp

@@ -40,4 +40,20 @@ PYBIND11_MODULE(_gc_mlir, m) {
         }
       },
       py::arg("context") = py::none(), py::arg("load") = true);
+
+  //===----------------------------------------------------------------------===//
+  // CPURuntime
+  //===----------------------------------------------------------------------===//
+  mlirRegisterCPURuntimePasses();
+  auto cpuruntimeM = m.def_submodule("cpuruntime");
+  cpuruntimeM.def(
+      "register_dialect",
+      [](MlirContext context, bool load) {
+        MlirDialectHandle dialect = mlirGetDialectHandle__cpuruntime__();
+        mlirDialectHandleRegisterDialect(dialect, context);
+        if (load) {
+          mlirDialectHandleLoadDialect(dialect, context);
+        }
+      },
+      py::arg("context") = py::none(), py::arg("load") = true);
 }

python/gc_mlir/_mlir_libs/_site_initialize_0.py

@@ -4,6 +4,8 @@
 
 
 def context_init_hook(context):
-    from ._gc_mlir.onednn_graph import register_dialect
-
-    register_dialect(context)
+    from ._gc_mlir.onednn_graph import register_dialect as register_onednn_graph_dialect
+    from ._gc_mlir.cpuruntime import register_dialect as register_cpuruntime_dialect
+    
+    register_onednn_graph_dialect(context)
+    register_cpuruntime_dialect(context)

python/gc_mlir/dialects/CPURuntimeOps.td

@@ -0,0 +1,6 @@
+#ifndef PYTHON_BINDINGS_CPURUNTIME_OPS
+#define PYTHON_BINDINGS_CPURUNTIME_OPS
+
+include "gc/Dialect/CPURuntime/IR/CPURuntimeOps.td"
+
+#endif // PYTHON_BINDINGS_CPURUNTIME_OPS

python/gc_mlir/dialects/OneDNNGraphOps.td

@@ -11,4 +11,4 @@
 
 include "gc/Dialect/OneDNNGraph/OneDNNGraphOps.td"
 
-#endif
+#endif // PYTHON_BINDINGS_ONEDNNGRAPH_OPS

python/gc_mlir/dialects/cpuruntime.py

@@ -0,0 +1,2 @@
+from ._cpuruntime_ops_gen import *
+from .._mlir_libs._gc_mlir.cpuruntime import *

test/gc/python/dialects/cpuruntime.py

@@ -0,0 +1,34 @@
+# See the License for the specific language governing permissions and
+# limitations under the License.
+# ===============================================================================
+# RUN: %python %s | FileCheck %s
+
+from gc_mlir.ir import *
+from gc_mlir.dialects import cpuruntime, func
+
+
+def run(f):
+    print("\nTEST:", f.__name__)
+    f()
+    return f
+
+
+# CHECK-LABEL: TEST: testCPURuntimeOps
+@run
+def testCPURuntimeOps():
+    with Context() as ctx, Location.unknown():
+        module = Module.create()
+        with InsertionPoint(module.body):
+
+            @func.FuncOp.from_py_func(F32Type.get(), IntegerType.get_signless(32))
+            def do_print(arg1, arg2):
+                cpuruntime.printf("Hello world %f %d", [arg1, arg2])
+                return
+
+        # CHECK-LABEL: func @do_print(
+        # CHECK-SAME:                  %[[ARG_0:.*]]: f32, %[[ARG_1:.*]]: i32) {
+        # CHECK:         cpuruntime.printf "Hello world %f %d" %[[ARG_0]], %[[ARG_1:.*]] : f32, i32
+        # CHECK:         return
+        # CHECK:       }
+        print(module)
+

test/gc/python/dialects/onednn_graph.py

@@ -0,0 +1,36 @@
+# See the License for the specific language governing permissions and
+# limitations under the License.
+# ===============================================================================
+# RUN: %python %s | FileCheck %s
+
+from gc_mlir.ir import *
+from gc_mlir.dialects import onednn_graph, func
+
+
+def run(f):
+    print("\nTEST:", f.__name__)
+    f()
+    return f
+
+
+# CHECK-LABEL: TEST: testOneDNNGraphOps
+@run
+def testOneDNNGraphOps():
+    with Context() as ctx, Location.unknown():
+        module = Module.create()
+        with InsertionPoint(module.body):
+            f32 = F32Type.get(ctx)
+            tensor_type = RankedTensorType.get([128, 128], f32)
+
+            @func.FuncOp.from_py_func(tensor_type, tensor_type)
+            def entry(arg1, arg2):
+                res1 = onednn_graph.matmul(
+                    arg1, arg2, bias=None, transpose_a=False, transpose_b=False
+                )
+                res2 = onednn_graph.add(res1, arg2)
+                return onednn_graph.relu(res2)
+
+        # CHECK: [[MM:%.+]] = onednn_graph.matmul
+        # CHECK: [[ADD:%.+]] = onednn_graph.add
+        # CHECK: [[RELU:%.+]] = onednn_graph.relu
+    print(module)

test/gc/Python/lit.local.cfg renamed to test/gc/python/lit.local.cfg

@@ -14,10 +14,10 @@ def run(f):
     return f
 
 
-# CHECK-LABEL: TEST: testCreatetOp
+# CHECK-LABEL: TEST: testCreateOp
 # CHECK onednn_graph.add
 @run
-def testCreatetOp():
+def testCreateOp():
     with Context() as ctx, Location.unknown():
         module = Module.create()
         f32 = F32Type.get(ctx)