[bazel] More fixes for 35df525

I missed a couple spots in my previous attempt to fix the build
(4eb7c34).

Author: slackito

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",
     ],