[llvm] [bazel] Fix AMXDialect (PR #115221)

[Pranav Kant via llvm-commits]

https://github.com/pranavk created https://github.com/llvm/llvm-project/pull/115221

None

>From 38892d42ed44fb7f5d8d3ccc132ac5c346bc88ff Mon Sep 17 00:00:00 2001
From: Pranav Kant <prka at google.com>
Date: Wed, 6 Nov 2024 21:57:02 +0000
Subject: [PATCH] [bazel] Fix AMXDialect

---
 .../bazel/llvm-project-overlay/mlir/BUILD.bazel  | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

diff --git a/utils/bazel/llvm-project-overlay/mlir/BUILD.bazel b/utils/bazel/llvm-project-overlay/mlir/BUILD.bazel
index e820234caa064a..b85e7413f91844 100644
--- a/utils/bazel/llvm-project-overlay/mlir/BUILD.bazel
+++ b/utils/bazel/llvm-project-overlay/mlir/BUILD.bazel
@@ -2360,6 +2360,20 @@ gentbl_cc_library(
             ],
             "include/mlir/Dialect/AMX/AMXDialect.cpp.inc",
         ),
+        (
+            [
+                "-gen-typedef-decls",
+                "-typedefs-dialect=amx",
+            ],
+            "include/mlir/Dialect/AMX/AMXTypes.h.inc",
+        ),
+        (
+            [
+                "-gen-typedef-defs",
+                "-typedefs-dialect=amx",
+            ],
+            "include/mlir/Dialect/AMX/AMXTypes.cpp.inc",
+        ),
         (
             ["-gen-op-decls"],
             "include/mlir/Dialect/AMX/AMX.h.inc",
@@ -2389,6 +2403,7 @@ cc_library(
         ":IR",
         ":LLVMDialect",
         ":SideEffectInterfaces",
+        "//llvm:Support",
     ],
 )
 
@@ -2399,6 +2414,7 @@ cc_library(
     includes = ["include"],
     deps = [
         ":AMXDialect",
+        ":ConvertToLLVMInterface",
         ":IR",
         ":LLVMCommonConversion",
         ":LLVMDialect",