From 67b02f487dc8ef8a6cddc1f968544e0b4cb37ad1 Mon Sep 17 00:00:00 2001
From: Rahul Joshi <jurahul@google.com>
Date: Fri, 26 Feb 2021 14:18:04 -0800
Subject: [PATCH] [XLA:GPU] Fix build failure

PiperOrigin-RevId: 359837344
Change-Id: I140aba2db41a860c1c5eb5dd191c3b066ef4fa01
---
 tensorflow/compiler/xla/service/mlir_gpu/kernel_lowering.cc | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/tensorflow/compiler/xla/service/mlir_gpu/kernel_lowering.cc b/tensorflow/compiler/xla/service/mlir_gpu/kernel_lowering.cc
index 22404cd1d54..3f547fe3308 100644
--- a/tensorflow/compiler/xla/service/mlir_gpu/kernel_lowering.cc
+++ b/tensorflow/compiler/xla/service/mlir_gpu/kernel_lowering.cc
@@ -215,7 +215,9 @@ class LowerToROCDLPass
     {
       ::mlir::OwningRewritePatternList patterns;
       ::mlir::populateGpuRewritePatterns(m.getContext(), patterns);
-      ::mlir::applyPatternsAndFoldGreedily(m, std::move(patterns));
+      if (failed(mlir::applyPatternsAndFoldGreedily(m, std::move(patterns)))) {
+        signalPassFailure();
+      }
     }
 
     ::mlir::OwningRewritePatternList patterns;