[llvm] [bazel][mlir][python] Port #157127: UB dialect filegroup (PR #157473)

[Jordan Rupprecht via llvm-commits]

https://github.com/rupprecht created https://github.com/llvm/llvm-project/pull/157473

Used downstream

>From 344f33dff1f504faae1e647f4e712b4670d7bb5d Mon Sep 17 00:00:00 2001
From: Jordan Rupprecht <rupprecht at google.com>
Date: Mon, 8 Sep 2025 07:25:07 -0700
Subject: [PATCH] [bazel][mlir][python] Port #157127: UB dialect filegroup

---
 .../mlir/python/BUILD.bazel                   | 25 +++++++++++++++++++
 1 file changed, 25 insertions(+)

diff --git a/utils/bazel/llvm-project-overlay/mlir/python/BUILD.bazel b/utils/bazel/llvm-project-overlay/mlir/python/BUILD.bazel
index 8114e128d615c..944a911bccc17 100644
--- a/utils/bazel/llvm-project-overlay/mlir/python/BUILD.bazel
+++ b/utils/bazel/llvm-project-overlay/mlir/python/BUILD.bazel
@@ -1428,6 +1428,31 @@ filegroup(
     srcs = glob(["mlir/dialects/transform/interpreter/*.py"]),
 )
 
+##---------------------------------------------------------------------------##
+# UB dialect.
+##---------------------------------------------------------------------------##
+
+gentbl_filegroup(
+    name = "UbOpsPyGen",
+    tbl_outs = {"mlir/dialects/_ub_ops_gen.py": [
+        "-gen-python-op-bindings",
+        "-bind-dialect=ub",
+    ]},
+    tblgen = "//mlir:mlir-tblgen",
+    td_file = "mlir/dialects/UBOps.td",
+    deps = [
+        "//mlir:UBDialectTdFiles",
+    ],
+)
+
+filegroup(
+    name = "UbPyFiles",
+    srcs = [
+        "mlir/dialects/ub.py",
+        ":UbOpsPyGen",
+    ],
+)
+
 ##---------------------------------------------------------------------------##
 # Vector dialect.
 ##---------------------------------------------------------------------------##