[mlir][complex] Add Python bindings for complex ops.

mlir/python/CMakeLists.txt

declare_mlir_dialect_python_bindings(
   SOURCES
     dialects/builtin.py
     dialects/_builtin_ops_ext.py
   DIALECT_NAME builtin)
 
 declare_mlir_dialect_python_bindings(
   ADD_TO_PARENT MLIRPythonSources.Dialects
   ROOT_DIR "${CMAKE_CURRENT_SOURCE_DIR}/mlir"
+  TD_FILE dialects/ComplexOps.td
+  SOURCES
+    dialects/complex.py
+  DIALECT_NAME complex)
+
+declare_mlir_dialect_python_bindings(
+  ADD_TO_PARENT MLIRPythonSources.Dialects
+  ROOT_DIR "${CMAKE_CURRENT_SOURCE_DIR}/mlir"
   TD_FILE dialects/ControlFlowOps.td
   SOURCES
     dialects/cf.py
   DIALECT_NAME cf)
 
 declare_mlir_dialect_python_bindings(
   ADD_TO_PARENT MLIRPythonSources.Dialects
   ROOT_DIR "${CMAKE_CURRENT_SOURCE_DIR}/mlir"

mlir/python/mlir/dialects/ComplexOps.td

+//===-- ComplexOps.td - Entry point for ComplexOps bindings ---------------===//
+//
+// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
+// See https://llvm.org/LICENSE.txt for license information.
+// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
+//
+//===----------------------------------------------------------------------===//
+
+#ifndef PYTHON_BINDINGS_COMPLEX_OPS
+#define PYTHON_BINDINGS_COMPLEX_OPS
+
+include "mlir/Bindings/Python/Attributes.td"
+include "mlir/Dialect/Complex/IR/ComplexOps.td"
+
+#endif

mlir/python/mlir/dialects/complex.py

+#  Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
+#  See https://llvm.org/LICENSE.txt for license information.
+#  SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
+
+from ._complex_ops_gen import *

mlir/test/python/dialects/complex_dialect.py

+# RUN: %PYTHON %s | FileCheck %s
+
+# Naming this file with a `_dialect` suffix to avoid a naming conflict with
+# python package's math module (coming in from random.py).
+
+from mlir.ir import *
+import mlir.dialects.func as func
+import mlir.dialects.complex as mlir_complex
+
+
+def run(f):
+  print("\nTEST:", f.__name__)
+  f()
+
+
+# CHECK-LABEL: TEST: testComplexOps
+@run
+def testComplexOps():
+  with Context() as ctx, Location.unknown():
+    module = Module.create()
+    with InsertionPoint(module.body):
+
+      @func.FuncOp.from_py_func(ComplexType.get(F32Type.get()))
+      def emit_add(arg):
+        return mlir_complex.AddOp(arg, arg)
+
+    # CHECK-LABEL: func @emit_add(
+    # CHECK-SAME:                  %[[ARG:.*]]: complex<f32>) -> complex<f32> {
+    # CHECK:         %[[RES:.*]] = complex.add %[[ARG]], %[[ARG]] : complex<f32>
+    # CHECK:         return %[[RES]] : complex<f32>
+    # CHECK:       }
+    print(module)

utils/bazel/llvm-project-overlay/mlir/python/BUILD.bazel

filegroup(
     srcs = [
         "mlir/dialects/_bufferization_ops_ext.py",
         "mlir/dialects/bufferization.py",
         ":BufferizationOpsPyGen",
     ],
 )
 
 ##---------------------------------------------------------------------------##
+# Complex dialect.
+##---------------------------------------------------------------------------##
+
+td_library(
+    name = "ComplexOpsPyTdFiles",
+    srcs = [
+        "//mlir:include/mlir/Bindings/Python/Attributes.td",
+    ],
+    includes = ["../include"],
+    deps = [
+        "//mlir:ComplexOpsTdFiles",
+        "//mlir:OpBaseTdFiles",
+    ],
+)
+
+gentbl_filegroup(
+    name = "ComplexOpsPyGen",
+    tbl_outs = [
+        (
+            [
+                "-gen-python-op-bindings",
+                "-bind-dialect=complex",
+            ],
+            "mlir/dialects/_complex_ops_gen.py",
+        ),
+    ],
+    tblgen = "//mlir:mlir-tblgen",
+    td_file = "mlir/dialects/ComplexOps.td",
+    deps = [
+        ":ComplexOpsPyTdFiles",
+    ],
+)
+
+filegroup(
+    name = "ComplexOpsPyFiles",
+    srcs = [
+        "mlir/dialects/complex.py",
+        ":ComplexOpsPyGen",
+    ],
+)
+
+##---------------------------------------------------------------------------##
 # ControlFlow dialect.
 ##---------------------------------------------------------------------------##
 
 td_library(
     name = "ControlFlowOpsPyTdFiles",
     srcs = [
         "//mlir:include/mlir/Bindings/Python/Attributes.td",
     ],