[llvm] Fix Bazel build (PR #222232)

[Anshul Nigham via llvm-commits]

https://github.com/nigham created https://github.com/llvm/llvm-project/pull/222232

Add missing deps and update sharding for ROCDLDialect

>From 8f1cd8856015961ee967bf5475b74cdd4c647cf5 Mon Sep 17 00:00:00 2001
From: Anshul Nigham <nigham at google.com>
Date: Tue, 8 Sep 2026 21:50:18 -0700
Subject: [PATCH] Fix Bazel build: add missing deps and update sharding for
 ROCDLDialect

---
 .../lib/Optimizer/OpenACC/Support/BUILD.bazel |  1 +
 .../Optimizer/OpenACC/Transforms/BUILD.bazel  |  3 +++
 .../llvm-project-overlay/mlir/BUILD.bazel     | 25 ++++++++++++++++---
 3 files changed, 25 insertions(+), 4 deletions(-)

diff --git a/utils/bazel/llvm-project-overlay/flang/lib/Optimizer/OpenACC/Support/BUILD.bazel b/utils/bazel/llvm-project-overlay/flang/lib/Optimizer/OpenACC/Support/BUILD.bazel
index 0198547e82853..40cac3bb4544c 100644
--- a/utils/bazel/llvm-project-overlay/flang/lib/Optimizer/OpenACC/Support/BUILD.bazel
+++ b/utils/bazel/llvm-project-overlay/flang/lib/Optimizer/OpenACC/Support/BUILD.bazel
@@ -45,6 +45,7 @@ cc_library(
         "//mlir:ArithDialect",
         "//mlir:ControlFlowInterfaces",
         "//mlir:IR",
+        "//mlir:LLVMDialect",
         "//mlir:OpenACCDialect",
         "//mlir:OpenACCUtils",
         "//mlir:Support",
diff --git a/utils/bazel/llvm-project-overlay/flang/lib/Optimizer/OpenACC/Transforms/BUILD.bazel b/utils/bazel/llvm-project-overlay/flang/lib/Optimizer/OpenACC/Transforms/BUILD.bazel
index 818d3fda8cb4f..96f397f659abb 100644
--- a/utils/bazel/llvm-project-overlay/flang/lib/Optimizer/OpenACC/Transforms/BUILD.bazel
+++ b/utils/bazel/llvm-project-overlay/flang/lib/Optimizer/OpenACC/Transforms/BUILD.bazel
@@ -24,10 +24,12 @@ cc_library(
     ),
     deps = [
         "//flang:RuntimeHeaders",
+        "//flang:SemanticsHeaders",
         "//flang/include/flang/Optimizer/Analysis:AnalysisHeaders",
         "//flang/include/flang/Optimizer/Builder:BuilderHeaders",
         "//flang/include/flang/Optimizer/Dialect:DialectHeaders",
         "//flang/include/flang/Optimizer/Dialect/CUF:CUFHeaders",
+        "//flang/include/flang/Optimizer/Dialect/CUF/Attributes:CUFAttrsHeaders",
         "//flang/include/flang/Optimizer/HLFIR:HLFIRHeaders",
         "//flang/include/flang/Optimizer/OpenACC:FIROpenACCPassesIncGen",
         "//flang/include/flang/Optimizer/OpenACC:OpenACCHeaders",
@@ -42,6 +44,7 @@ cc_library(
         "//flang/lib/Optimizer/OpenACC/Support:FIROpenACCSupport",
         "//llvm:Support",
         "//mlir:Analysis",
+        "//mlir:ArithDialect",
         "//mlir:FuncDialect",
         "//mlir:IR",
         "//mlir:OpenACCAnalysis",
diff --git a/utils/bazel/llvm-project-overlay/mlir/BUILD.bazel b/utils/bazel/llvm-project-overlay/mlir/BUILD.bazel
index a6c600f76de18..69b557d77a910 100644
--- a/utils/bazel/llvm-project-overlay/mlir/BUILD.bazel
+++ b/utils/bazel/llvm-project-overlay/mlir/BUILD.bazel
@@ -6708,9 +6708,19 @@ cc_library(
 
 cc_library(
     name = "ROCDLDialect",
-    srcs = ["lib/Dialect/LLVMIR/IR/ROCDLDialect.cpp"],
+    srcs = [
+        "lib/Dialect/LLVMIR/IR/ROCDLDialect.cpp",
+        "lib/Dialect/LLVMIR/IR/ROCDLOps.h",
+    ] + gentbl_sharded_op_defs(
+        name = "ROCDLOps_sharded_defs",
+        shard_count = 8,
+        source_file = "lib/Dialect/LLVMIR/IR/ROCDLOps.cpp",
+    ),
     hdrs = ["include/mlir/Dialect/LLVMIR/ROCDLDialect.h"],
-    includes = ["include"],
+    includes = [
+        "include",
+        "lib/Dialect/LLVMIR",
+    ],
     deps = [
         ":BytecodeOpInterface",
         ":GPUDialect",
@@ -6778,8 +6788,14 @@ td_library(
 gentbl_cc_library(
     name = "ROCDLOpsIncGen",
     tbl_outs = {
-        "include/mlir/Dialect/LLVMIR/ROCDLOps.h.inc": ["-gen-op-decls"],
-        "include/mlir/Dialect/LLVMIR/ROCDLOps.cpp.inc": ["-gen-op-defs"],
+        "include/mlir/Dialect/LLVMIR/ROCDLOps.h.inc": [
+            "-gen-op-decls",
+            "-op-shard-count=8",
+        ],
+        "include/mlir/Dialect/LLVMIR/ROCDLOps.cpp.inc": [
+            "-gen-op-defs",
+            "-op-shard-count=8",
+        ],
         "include/mlir/Dialect/LLVMIR/ROCDLOpsDialect.h.inc": [
             "-gen-dialect-decls",
             "-dialect=rocdl",
@@ -10586,6 +10602,7 @@ td_library(
         "include/mlir/Dialect/OpenACC/OpenACCAttributes.td",
         "include/mlir/Dialect/OpenACC/OpenACCBase.td",
         "include/mlir/Dialect/OpenACC/OpenACCCGAttributes.td",
+        "include/mlir/Dialect/OpenACC/OpenACCCGEnums.td",
         "include/mlir/Dialect/OpenACC/OpenACCCGOps.td",
         "include/mlir/Dialect/OpenACC/OpenACCOps.td",
         "include/mlir/Dialect/OpenACC/OpenACCOpsInterfaces.td",