[llvm] [mlir] [MLIR][Python] Add shard Dialect Python Bindings (PR #162578)

Siavash Nazari via llvm-commits llvm-commits at lists.llvm.org
Thu Oct 9 12:27:35 PDT 2025


================
@@ -981,6 +981,38 @@ filegroup(
     ],
 )
 
+##---------------------------------------------------------------------------##
+# Shard dialect.
+##---------------------------------------------------------------------------##
+
+gentbl_filegroup(
+    name = "ShardOpsPyGen",
+    tbl_outs = {
+        "mlir/dialects/_shard_enum_gen.py": [
+            "-gen-python-enum-bindings",
+            "-bind-dialect=shard",
+        ],
+        "mlir/dialects/_shard_ops_gen.py": [
+            "-gen-python-op-bindings",
+            "-bind-dialect=shard",
+        ],
+    },
+    tblgen = "//mlir:mlir-tblgen",
+    td_file = "mlir/dialects/ShardOps.td",
+    deps = [
+        "//mlir:OpBaseTdFiles",
+        "//mlir:ShardTdFiles",
+    ],
+)
+
+filegroup(
+    name = "ShardOpsPyFiles",
+    srcs = [
+        "mlir/dialects/shard.py",
+        ":ShardOpsPyGen",
+    ],
+)
----------------
Svoch wrote:

Thanks @makslevental! Seems like Bazel build is fine. Can I get reviews please? @jpienaar @superbobry 

https://github.com/llvm/llvm-project/pull/162578


More information about the llvm-commits mailing list