a.diff

diff --git a/compiler/src/iree/compiler/API/BUILD.bazel b/compiler/src/iree/compiler/API/BUILD.bazel
index 543a088fba..43ff10395d 100644
--- a/compiler/src/iree/compiler/API/BUILD.bazel
+++ b/compiler/src/iree/compiler/API/BUILD.bazel
@@ -38,6 +38,7 @@ iree_compiler_cc_library(
         "//compiler/src/iree/compiler/API/Internal:IREEReduceToolEntryPoint",
         "//compiler/src/iree/compiler/API/Internal:LLDToolEntryPoint",
         "//llvm-external-projects/iree-dialects:CAPI",
+        "@llvm-project//mlir:CAPIAMDGPU",
         "@llvm-project//mlir:CAPIDebug",
         "@llvm-project//mlir:CAPIGPU",
         "@llvm-project//mlir:CAPIIR",
diff --git a/compiler/src/iree/compiler/API/CMakeLists.txt b/compiler/src/iree/compiler/API/CMakeLists.txt
index f28498a3e4..15cf1f955b 100644
--- a/compiler/src/iree/compiler/API/CMakeLists.txt
+++ b/compiler/src/iree/compiler/API/CMakeLists.txt
@@ -15,6 +15,7 @@ iree_cc_library(
     StaticImpl
   DEPS
     IREEDialectsCAPI
+    MLIRCAPIAMDGPU
     MLIRCAPIDebug
     MLIRCAPIGPU
     MLIRCAPIIR
diff --git a/compiler/src/iree/compiler/API/Internal/BUILD.bazel b/compiler/src/iree/compiler/API/Internal/BUILD.bazel
index 11db760442..36fb9997b3 100644
--- a/compiler/src/iree/compiler/API/Internal/BUILD.bazel
+++ b/compiler/src/iree/compiler/API/Internal/BUILD.bazel
@@ -162,6 +162,7 @@ iree_compiler_cc_library(
         "//compiler/src/iree/compiler/Codegen/Utils",
         "//compiler/src/iree/compiler/Dialect/LinalgExt/IR",
         "//compiler/src/iree/compiler/Dialect/LinalgExt/Utils",
+        "@llvm-project//mlir:CAPIAMDGPU",
         "@llvm-project//mlir:CAPIIR",
         "@llvm-project//mlir:CAPIIRHeaders",
         "@llvm-project//mlir:DialectUtils",
@@ -180,6 +181,7 @@ iree_compiler_cc_library(
         "//compiler/bindings/c:headers",
         "//compiler/src/iree/compiler/Codegen/Dialect/GPU/IR:IREEGPUDialect",
         "//compiler/src/iree/compiler/Codegen/Utils",
+        "@llvm-project//mlir:CAPIAMDGPU",
         "@llvm-project//mlir:CAPIIR",
         "@llvm-project//mlir:CAPIIRHeaders",
         "@llvm-project//mlir:IR",
diff --git a/compiler/src/iree/compiler/API/Internal/CMakeLists.txt b/compiler/src/iree/compiler/API/Internal/CMakeLists.txt
index b68b35a226..27bd2c4404 100644
--- a/compiler/src/iree/compiler/API/Internal/CMakeLists.txt
+++ b/compiler/src/iree/compiler/API/Internal/CMakeLists.txt
@@ -135,6 +135,7 @@ iree_cc_library(
     "IREECodegenDialectCAPI.cpp"
   DEPS
     IREELLVMIncludeSetup
+    MLIRCAPIAMDGPU
     MLIRCAPIIR
     MLIRIR
     MLIRLinalgDialect
@@ -154,6 +155,7 @@ iree_cc_library(
     "IREEGPUDialectCAPI.cpp"
   DEPS
     IREELLVMIncludeSetup
+    MLIRCAPIAMDGPU
     MLIRCAPIIR
     MLIRIR
     iree::compiler::Codegen::Dialect::GPU::IR::IREEGPUDialect
diff --git a/llvm-external-projects/iree-dialects/BUILD.bazel b/llvm-external-projects/iree-dialects/BUILD.bazel
index 2dd253482e..16afefe3a6 100644
--- a/llvm-external-projects/iree-dialects/BUILD.bazel
+++ b/llvm-external-projects/iree-dialects/BUILD.bazel
@@ -236,6 +236,7 @@ cc_library(
     deps = [
         ":IREELinalgTransformDialect",
         ":IREELinalgTransformDialectPasses",
+        "@llvm-project//mlir:CAPIAMDGPU",
         "@llvm-project//mlir:CAPIIR",
         "@llvm-project//mlir:IR",
         "@llvm-project//mlir:LinalgTransformOps",