Add mhlo python binding generator target

This just invokes the generator backend & creates a filegroup.

PiperOrigin-RevId: 377318653

BUILD

@@ -1250,3 +1250,43 @@ cc_binary(
         "@llvm-project//mlir:Support",
     ],
 )
+
+# Python library.
+
+td_library(
+    name = "MhloOpsPyTdFiles",
+    srcs = [
+        "@llvm-project//mlir:include/mlir/Bindings/Python/Attributes.td",
+    ],
+    includes = ["include"],
+    deps = [
+        ":hlo_ops_td_files",
+        "@llvm-project//mlir:OpBaseTdFiles",
+    ],
+)
+
+gentbl_filegroup(
+    name = "MhloOpsPyGen",
+    tbl_outs = [
+        (
+            [
+                "-gen-python-op-bindings",
+                "-bind-dialect=mhlo",
+            ],
+            "python/_mhlo_ops_gen.py",
+        ),
+    ],
+    tblgen = "@llvm-project//mlir:mlir-tblgen",
+    td_file = "python/MhloOps.td",
+    deps = [
+        ":MhloOpsPyTdFiles",
+    ],
+)
+
+filegroup(
+    name = "MhloOpsPyFiles",
+    srcs = [
+        "python/mhlo.py",
+        ":MhloOpsPyGen",
+    ],
+)

CMakeLists.txt

@@ -85,5 +85,6 @@ add_custom_target(check-mlir-hlo)
 
 add_subdirectory(include/mlir-hlo)
 add_subdirectory(lib)
+add_subdirectory(python)
 add_subdirectory(tools)
 add_subdirectory(tests)

python/CMakeLists.txt

@@ -0,0 +1,9 @@
+include(AddMLIRPython)
+
+################################################################################
+# Generate dialect-specific bindings.
+################################################################################
+
+add_mlir_dialect_python_bindings(MLIRBindingsPythonMhloOps
+  TD_FILE MhloOps.td
+  DIALECT_NAME mhlo_dialect)

python/MhloOps.td

@@ -0,0 +1,24 @@
+/* Copyright 2021 The TensorFlow Authors. All Rights Reserved.
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+==============================================================================*/
+
+// Entry point for MHLO ops bind.
+
+#ifndef PYTHON_BINDINGS_MHLO_OPS
+#define PYTHON_BINDINGS_MHLO_OPS
+
+include "mlir/Bindings/Python/Attributes.td"
+include "mlir-hlo/Dialect/mhlo/IR/hlo_ops.td"
+
+#endif

python/mhlo.py

@@ -0,0 +1,18 @@
+# Copyright 2020 The TensorFlow Authors. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+# ==============================================================================
+
+# pylint: disable=wildcard-import,relative-beyond-top-level
+from ._mhlo_ops_gen import *
+# pylint: enable=wildcard-import,relative-beyond-top-level