diff --git a/utils/bazel/llvm-project-overlay/mlir/BUILD.bazel b/utils/bazel/llvm-project-overlay/mlir/BUILD.bazel
--- a/utils/bazel/llvm-project-overlay/mlir/BUILD.bazel
+++ b/utils/bazel/llvm-project-overlay/mlir/BUILD.bazel
@@ -6421,6 +6421,7 @@
         "//llvm:Support",
         "//mlir/test:TestAffine",
         "//mlir/test:TestAnalysis",
+        "//mlir/test:TestArithmetic",
         "//mlir/test:TestDLTI",
         "//mlir/test:TestDialect",
         "//mlir/test:TestFunc",
diff --git a/utils/bazel/llvm-project-overlay/mlir/test/BUILD.bazel b/utils/bazel/llvm-project-overlay/mlir/test/BUILD.bazel
--- a/utils/bazel/llvm-project-overlay/mlir/test/BUILD.bazel
+++ b/utils/bazel/llvm-project-overlay/mlir/test/BUILD.bazel
@@ -619,6 +619,23 @@
     ],
 )
 
+cc_library(
+    name = "TestArithmetic",
+    srcs = glob(["lib/Dialect/Arithmetic/*.cpp"]),
+    defines = ["MLIR_CUDA_CONVERSIONS_ENABLED"],
+    includes = ["lib/Dialect/Test"],
+    deps = [
+        "//mlir:ArithmeticDialect",
+        "//mlir:ArithmeticTransforms",
+        "//mlir:FuncDialect",
+        "//mlir:LLVMDialect",
+        "//mlir:MathDialect",
+        "//mlir:Pass",
+        "//mlir:Transforms",
+        "//mlir:VectorDialect",
+    ],
+)
+
 cc_library(
     name = "TestShapeDialect",
     srcs = [