experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

[XLA:GPU] Fix build failure

Browse Source 
PiperOrigin-RevId: 359837344
Change-Id: I140aba2db41a860c1c5eb5dd191c3b066ef4fa01
This commit is contained in:
Rahul Joshi 2021-02-26 14:18:04 -08:00 committed by TensorFlower Gardener
parent 03e82caf2e
commit 67b02f487d
1 changed files with 3 additions and 1 deletions

4
tensorflow/compiler/xla/service/mlir_gpu/kernel_lowering.cc
View File

@ -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;