[bazel] Slight polish after ef68a53.

chsigg · chsigg · commit 48f53422fe7f · 2026-02-09T19:38:22.000+01:00
diff --git a/utils/bazel/llvm-project-overlay/mlir/BUILD.bazel b/utils/bazel/llvm-project-overlay/mlir/BUILD.bazel
@@ -714,14 +714,10 @@ mlir_c_api_cc_library(
     name = "CAPIMath",
     srcs = ["lib/CAPI/Dialect/Math.cpp"],
     hdrs = ["include/mlir-c/Dialect/Math.h"],
-    capi_deps = [
-        ":CAPIIR",
-    ],
+    capi_deps = [":CAPIIR"],
+    header_deps = [":MathPassIncGen"],
     includes = ["include"],
-    deps = [
-        ":MathDialect",
-        ":MathPassIncGen",
-    ],
+    deps = [":MathDialect"],
 )
 
 mlir_c_api_cc_library(
@@ -856,14 +852,10 @@ mlir_c_api_cc_library(
     name = "CAPISCF",
     srcs = ["lib/CAPI/Dialect/SCF.cpp"],
     hdrs = ["include/mlir-c/Dialect/SCF.h"],
-    capi_deps = [
-        ":CAPIIR",
-    ],
+    capi_deps = [":CAPIIR"],
+    header_deps = [":SCFPassIncGen"],
     includes = ["include"],
-    deps = [
-        ":SCFDialect",
-        ":SCFPassIncGen",
-    ],
+    deps = [":SCFDialect"],
 )
 
 mlir_c_api_cc_library(
@@ -1430,7 +1422,6 @@ py_binary(
     name = "stubgen_runner",
     srcs = ["stubgen_runner.py"],
     data = ["@nanobind//:src/stubgen.py"],
-    visibility = ["//visibility:private"],
     deps = ["@rules_python//python/runfiles"],
 )
 
diff --git a/utils/bazel/llvm-project-overlay/mlir/build_defs.bzl b/utils/bazel/llvm-project-overlay/mlir/build_defs.bzl
@@ -73,7 +73,7 @@ def mlir_c_api_cc_library(
         **kwargs
     )
 
-def nanobind_pyi_genrule(name, module_name, outs, deps, visibility = None):
+def nanobind_pyi_genrule(name, module_name, outs, deps, **kwargs):
     """Generates .pyi stub file(s) for a nanobind extension module.
 
     Args:
@@ -91,5 +91,5 @@ def nanobind_pyi_genrule(name, module_name, outs, deps, visibility = None):
         outs = outs,
         cmd = "$(location :stubgen_runner) --module " + module_name + " --deps " + deps_arg + " -o $(RULEDIR)",
         tools = [":stubgen_runner"],
-        visibility = visibility,
+        **kwargs
     )

Original file line number	Diff line number	Diff line change
`@@ -73,7 +73,7 @@ def mlir_c_api_cc_library(`
`73`	`73`	`**kwargs`
`74`	`74`	`)`
`75`	`75`
`76`		`-def nanobind_pyi_genrule(name, module_name, outs, deps, visibility = None):`
	`76`	`+def nanobind_pyi_genrule(name, module_name, outs, deps, **kwargs):`
`77`	`77`	`"""Generates .pyi stub file(s) for a nanobind extension module.`
`78`	`78`
`79`	`79`	`Args:`
`@@ -91,5 +91,5 @@ def nanobind_pyi_genrule(name, module_name, outs, deps, visibility = None):`
`91`	`91`	`outs = outs,`
`92`	`92`	`cmd = "$(location :stubgen_runner) --module " + module_name + " --deps " + deps_arg + " -o $(RULEDIR)",`
`93`	`93`	`tools = [":stubgen_runner"],`
`94`		`- visibility = visibility,`
	`94`	`+ **kwargs`
`95`	`95`	`)`