NFC: Add @llvm-project//mlir:TransformUtils dependency as preparation for llvm/llvm-project#86819.

chsigg · copybara-github · commit 60a6790979bc · 2024-03-28T11:47:48.000-07:00
PiperOrigin-RevId: 620006777
diff --git a/third_party/triton/cl619443019.patch b/third_party/triton/cl619443019.patch
@@ -0,0 +1,76 @@
+==== triton/BUILD#44 - /google/src/cloud/csigg/mlir_transform_utils/triton/BUILD ====
+# action=edit type=text
+--- triton/BUILD	2024-03-22 08:02:38.000000000 -0700
++++ triton/BUILD	2024-03-27 01:34:43.000000000 -0700
+@@ -620,6 +620,7 @@
+         "@llvm-project//mlir:FunctionInterfaces",
+         "@llvm-project//mlir:GPUDialect",
+         "@llvm-project//mlir:IR",
++        "@llvm-project//mlir:InliningUtils",
+         "@llvm-project//mlir:LLVMDialect",
+         "@llvm-project//mlir:MathDialect",
+         "@llvm-project//mlir:SCFDialect",
+@@ -628,6 +629,7 @@
+         # The following is added to make Utility compile
+         ":TritonTools",
+         "@llvm-project//mlir:LLVMCommonConversion",
++        "@llvm-project//mlir:TransformUtils",
+         "@llvm-project//mlir:Transforms",
+     ],
+ )
+@@ -646,6 +648,7 @@
+         "@llvm-project//mlir:IR",
+         "@llvm-project//mlir:Pass",
+         "@llvm-project//mlir:Support",
++        "@llvm-project//mlir:TransformUtils",
+         "@llvm-project//mlir:Transforms",
+     ],
+     alwayslink = True,  # TritonDialect uses getCanonicalizationPatterns().
+@@ -729,6 +732,7 @@
+         "@llvm-project//mlir:LLVMDialect",
+         "@llvm-project//mlir:NVVMDialect",
+         "@llvm-project//mlir:Pass",
++        "@llvm-project//mlir:TransformUtils",
+         "@llvm-project//mlir:Transforms",
+     ],
+ )
+@@ -780,6 +784,7 @@
+         "@llvm-project//mlir:IR",
+         "@llvm-project//mlir:IndexDialect",
+         "@llvm-project//mlir:Pass",
++        "@llvm-project//mlir:TransformUtils",
+         "@llvm-project//mlir:Transforms",
+     ],
+ )
+==== triton/test/BUILD#18 - /google/src/cloud/csigg/mlir_transform_utils/triton/test/BUILD ====
+# action=edit type=text
+--- triton/test/BUILD	2024-03-11 11:42:57.000000000 -0700
++++ triton/test/BUILD	2024-03-27 01:32:04.000000000 -0700
+@@ -53,6 +53,7 @@
+         "@llvm-project//mlir:IR",
+         "@llvm-project//mlir:Pass",
+         "@llvm-project//mlir:SCFToControlFlow",
++        "@llvm-project//mlir:TransformUtils",
+         "@llvm-project//mlir:Transforms",
+         "//:TritonAnalysis",
+         "//:TritonDialects",
+==== triton/third_party/nvidia/BUILD#3 - /google/src/cloud/csigg/mlir_transform_utils/triton/third_party/nvidia/BUILD ====
+# action=edit type=text
+--- triton/third_party/nvidia/BUILD	2024-03-11 11:42:57.000000000 -0700
++++ triton/third_party/nvidia/BUILD	2024-03-27 01:32:46.000000000 -0700
+@@ -66,6 +66,7 @@
+         "@llvm-project//mlir:NVVMDialect",
+         "@llvm-project//mlir:Pass",
+         "@llvm-project//mlir:Support",
++        "@llvm-project//mlir:TransformUtils",
+         "@llvm-project//mlir:Transforms",
+         "//:TritonDialects",
+     ],
+@@ -113,6 +114,7 @@
+         "@llvm-project//mlir:NVVMDialect",
+         "@llvm-project//mlir:Pass",
+         "@llvm-project//mlir:SCFToControlFlow",
++        "@llvm-project//mlir:TransformUtils",
+         "@llvm-project//mlir:Transforms",
+         "//:TritonAnalysis",
+         "//:TritonDialects",
diff --git a/third_party/triton/workspace.bzl b/third_party/triton/workspace.bzl
@@ -17,5 +17,6 @@ def repo():
             "//third_party/triton:cl607293980.patch",  # long standing :(
             "//third_party/triton:cl617812302.patch",
             "//third_party/triton:cl619146327.patch",
+            "//third_party/triton:cl619443019.patch",
         ],
     )
diff --git a/xla/mlir/runtime/transforms/BUILD b/xla/mlir/runtime/transforms/BUILD
@@ -80,6 +80,7 @@ cc_library(
     deps = [
         "//xla/mlir/runtime/ir:rt",
         "@llvm-project//mlir:IR",
+        "@llvm-project//mlir:TransformUtils",
         "@llvm-project//mlir:Transforms",
     ],
 )
@@ -93,6 +94,7 @@ xla_cc_test(
         "//xla/mlir/runtime/ir:rt",
         "@llvm-project//mlir:IR",
         "@llvm-project//mlir:MemRefDialect",
+        "@llvm-project//mlir:TransformUtils",
         "@llvm-project//mlir:Transforms",
         "@tsl//tsl/platform:test",
         "@tsl//tsl/platform:test_main",
@@ -170,6 +172,7 @@ cc_library(
     deps = [
         ":custom_call_encoding",
         "//xla/runtime:type_id",
+        "@llvm-project//mlir:TransformUtils",
         "@llvm-project//mlir:Transforms",
     ],
 )
diff --git a/xla/mlir_hlo/BUILD b/xla/mlir_hlo/BUILD
@@ -306,6 +306,7 @@ cc_library(
         "@llvm-project//mlir:Pass",
         "@llvm-project//mlir:SCFDialect",
         "@llvm-project//mlir:Support",
+        "@llvm-project//mlir:TransformUtils",
         "@llvm-project//mlir:Transforms",
     ],
 )
@@ -446,6 +447,7 @@ cc_library(
         "@llvm-project//mlir:FunctionInterfaces",
         "@llvm-project//mlir:IR",
         "@llvm-project//mlir:InferTypeOpInterface",
+        "@llvm-project//mlir:InliningUtils",
         "@llvm-project//mlir:LLVMDialect",
         "@llvm-project//mlir:MemRefDialect",
         "@llvm-project//mlir:Pass",
@@ -619,6 +621,7 @@ cc_library(
         "@llvm-project//mlir:Support",
         "@llvm-project//mlir:TensorDialect",
         "@llvm-project//mlir:TensorUtils",
+        "@llvm-project//mlir:TransformUtils",
         "@llvm-project//mlir:Transforms",
         "@stablehlo//:base",
         "@stablehlo//:chlo_ops",
@@ -638,6 +641,7 @@ cc_library(
         "@llvm-project//mlir:FuncTransforms",
         "@llvm-project//mlir:IR",
         "@llvm-project//mlir:TensorDialect",
+        "@llvm-project//mlir:TransformUtils",
         "@llvm-project//mlir:Transforms",
         "@stablehlo//:stablehlo_ops",
     ],
@@ -791,6 +795,7 @@ cc_library(
         "@llvm-project//mlir:Support",
         "@llvm-project//mlir:TensorDialect",
         "@llvm-project//mlir:TensorUtils",
+        "@llvm-project//mlir:TransformUtils",
         "@llvm-project//mlir:Transforms",
         "@stablehlo//:chlo_ops",
     ],
@@ -878,6 +883,7 @@ cc_library(
         "@llvm-project//mlir:IR",
         "@llvm-project//mlir:ShapeDialect",
         "@llvm-project//mlir:TensorDialect",
+        "@llvm-project//mlir:TransformUtils",
         "@llvm-project//mlir:Transforms",
     ],
 )
@@ -909,6 +915,7 @@ cc_library(
         "@llvm-project//mlir:FuncDialect",
         "@llvm-project//mlir:IR",
         "@llvm-project//mlir:Support",
+        "@llvm-project//mlir:TransformUtils",
         "@llvm-project//mlir:Transforms",
         "@stablehlo//:stablehlo_ops",
         "@stablehlo//:stablehlo_ops_inc_gen",
@@ -928,6 +935,7 @@ cc_library(
         "@llvm-project//mlir:FuncDialect",
         "@llvm-project//mlir:IR",
         "@llvm-project//mlir:Support",
+        "@llvm-project//mlir:TransformUtils",
         "@llvm-project//mlir:Transforms",
         "@stablehlo//:stablehlo_ops",
         "@stablehlo//:stablehlo_ops_inc_gen",
@@ -1115,6 +1123,7 @@ cc_library(
         "@llvm-project//mlir:SCFTransforms",
         "@llvm-project//mlir:ShapeToStandard",
         "@llvm-project//mlir:TensorInferTypeOpInterfaceImpl",
+        "@llvm-project//mlir:TransformUtils",
         "@llvm-project//mlir:Transforms",
         "@llvm-project//mlir:VectorDialect",
         "@llvm-project//mlir:VectorToLLVM",

Original file line number	Diff line number	Diff line change
`@@ -17,5 +17,6 @@ def repo():`
`17`	`17`	`"//third_party/triton:cl607293980.patch", # long standing :(`
`18`	`18`	`"//third_party/triton:cl617812302.patch",`
`19`	`19`	`"//third_party/triton:cl619146327.patch",`
	`20`	`+ "//third_party/triton:cl619443019.patch",`
`20`	`21`	`],`
`21`	`22`	`)`