diff --git a/mlir/include/mlir/Dialect/SPIRV/Transforms/Passes.h b/mlir/include/mlir/Dialect/SPIRV/Transforms/Passes.h
--- a/mlir/include/mlir/Dialect/SPIRV/Transforms/Passes.h
+++ b/mlir/include/mlir/Dialect/SPIRV/Transforms/Passes.h
@@ -13,6 +13,7 @@
 #ifndef MLIR_DIALECT_SPIRV_TRANSFORMS_PASSES_H_
 #define MLIR_DIALECT_SPIRV_TRANSFORMS_PASSES_H_
 
+#include "mlir/Dialect/SPIRV/IR/SPIRVAttributes.h"
 #include "mlir/Pass/Pass.h"
 
 namespace mlir {
@@ -22,7 +23,6 @@
 namespace spirv {
 
 class ModuleOp;
-class TargetEnvAttr;
 
 //===----------------------------------------------------------------------===//
 // Passes