From 8df8252b98f2bd16e858b959ba20d122d83f37dd Mon Sep 17 00:00:00 2001
From: Danial Klimkin <dklimkin@google.com>
Date: Thu, 23 Jan 2025 15:15:44 +0100
Subject: [PATCH] [bazel]Fix(2) bazel build past
 2e6cc79f816d942ab09d6a310cd925c1da148aa9

---
 .../llvm-project-overlay/mlir/BUILD.bazel     | 39 +++++++++++++++----
 1 file changed, 32 insertions(+), 7 deletions(-)

diff --git a/utils/bazel/llvm-project-overlay/mlir/BUILD.bazel b/utils/bazel/llvm-project-overlay/mlir/BUILD.bazel
index d9f222982bc01..72c28faed1d16 100644
--- a/utils/bazel/llvm-project-overlay/mlir/BUILD.bazel
+++ b/utils/bazel/llvm-project-overlay/mlir/BUILD.bazel
@@ -5549,7 +5549,11 @@ cc_library(
         [
             "lib/Dialect/LLVMIR/Transforms/*.cpp",
         ],
-        exclude = ["lib/Dialect/LLVMIR/Transforms/LegalizeForExport.cpp"],
+        exclude = [
+            "lib/Dialect/LLVMIR/Transforms/DIExpressionLegalization.cpp",
+            "lib/Dialect/LLVMIR/Transforms/DIExpressionRewriter.cpp",
+            "lib/Dialect/LLVMIR/Transforms/LegalizeForExport.cpp",
+        ],
     ),
     hdrs = glob(
         [
@@ -5583,20 +5587,40 @@ cc_library(
 cc_library(
     name = "LLVMIRTransformsLegalizeForExport",
     srcs = ["lib/Dialect/LLVMIR/Transforms/LegalizeForExport.cpp"],
-    hdrs = [
-        "include/mlir/Dialect/LLVMIR/Transforms/DIExpressionLegalization.h",
-        "include/mlir/Dialect/LLVMIR/Transforms/DIExpressionRewriter.h",
-        "include/mlir/Dialect/LLVMIR/Transforms/LegalizeForExport.h",
-    ],
+    hdrs = ["include/mlir/Dialect/LLVMIR/Transforms/LegalizeForExport.h"],
     includes = ["include"],
     deps = [
         ":IR",
-        ":LLVMPassIncGen",
         ":LLVMDialect",
+        ":LLVMPassIncGen",
+        ":LLVMIRTransformsDIExpressionLegalization",
         ":Pass",
     ],
 )
 
+cc_library(
+    name = "LLVMIRTransformsDIExpressionLegalization",
+    srcs = ["lib/Dialect/LLVMIR/Transforms/DIExpressionLegalization.cpp"],
+    hdrs = ["include/mlir/Dialect/LLVMIR/Transforms/DIExpressionLegalization.h"],
+    includes = ["include"],
+    deps = [
+        ":LLVMIRTransformsDIExpressionRewriter",
+        "//llvm:BinaryFormat",
+    ],
+)
+
+cc_library(
+    name = "LLVMIRTransformsDIExpressionRewriter",
+    srcs = ["lib/Dialect/LLVMIR/Transforms/DIExpressionRewriter.cpp"],
+    hdrs = ["include/mlir/Dialect/LLVMIR/Transforms/DIExpressionRewriter.h"],
+    includes = ["include"],
+    deps = [
+        ":LLVMDialect",
+        ":TransformUtils",
+        "//llvm:Support",
+    ],
+)
+
 td_library(
     name = "GPUOpsTdFiles",
     srcs = [
@@ -9244,6 +9268,7 @@ cc_library(
         ":LLVMConversionIncGen",
         ":LLVMDialect",
         ":LLVMIntrinsicConversionIncGen",
+        ":LLVMIRTransformsDIExpressionLegalization",
         ":LLVMIRTransformsLegalizeForExport",
         ":OpenMPDialect",
         ":Support",