[bazel] More fixes for 35df525

I missed a couple spots in my previous attempt to fix the build (4eb7c34).
llvm · Jan 28, 2025 · 3a1e157 · 3a1e157
1 parent 6bb6c30
commit 3a1e157
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/utils/bazel/llvm-project-overlay/mlir/BUILD.bazel b/utils/bazel/llvm-project-overlay/mlir/BUILD.bazel
@@ -5109,6 +5109,7 @@ cc_library(
         ":SubsetOpInterface",
         ":Support",
         ":TensorDialect",
+        ":UBDialect",
         ":ValueBoundsOpInterface",
         ":VectorAttributesIncGen",
         ":VectorDialectIncGen",
@@ -6198,6 +6199,7 @@ cc_library(
         ":SPIRVDialect",
         ":Support",
         ":TransformUtils",
+        ":UBDialect",
         ":VectorDialect",
         "//llvm:Support",
     ],