Fix TFRT Kokoro build error related to mlir::LogicalResult.

PiperOrigin-RevId: 649259713
tensorflow · Jul 4, 2024 · 60277ba · 60277ba
1 parent 816371e
commit 60277ba
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 4 deletions.
diff --git a/BUILD b/BUILD
@@ -721,6 +721,7 @@ tfrt_cc_library(
         ":support",
         "@llvm-project//llvm:Support",
         "@llvm-project//mlir:IR",
+        "@llvm-project//mlir:Support",
         "@llvm-project//mlir:TranslateLib",
     ],
 )

diff --git a/include/tfrt/bef_converter/mlir_to_bef_translate.h b/include/tfrt/bef_converter/mlir_to_bef_translate.h
@@ -19,6 +19,7 @@
 #ifndef TFRT_BEF_CONVERTER_MLIR_TO_BEF_TRANSLATE_H_
 #define TFRT_BEF_CONVERTER_MLIR_TO_BEF_TRANSLATE_H_
 
+#include "mlir/Support/LogicalResult.h"
 #include "tfrt/support/forward_decls.h"
 
 namespace mlir {
@@ -27,9 +28,8 @@ class ModuleOp;
 
 namespace tfrt {
 
-llvm::LogicalResult MLIRToBEFTranslate(mlir::ModuleOp module,
+mlir::LogicalResult MLIRToBEFTranslate(mlir::ModuleOp module,
                                        llvm::raw_ostream& output);
-
 }
 
 #endif  // TFRT_BEF_CONVERTER_MLIR_TO_BEF_TRANSLATE_H_
diff --git a/include/tfrt/support/forward_decls.h b/include/tfrt/support/forward_decls.h
@@ -25,15 +25,13 @@
 
 #include "llvm/Support/Casting.h"
 #include "llvm/Support/ErrorHandling.h"
-#include "llvm/Support/LogicalResult.h"
 
 // We don't forward declare:
 //   DenseMap, SmallVector, StringMap, StringSet
 // Because they use default template parameters.
 namespace llvm {
 
 class raw_ostream;
-struct LogicalResult;
 
 template <typename T>
 class ArrayRef;