From 75dbd3373301d02d56c37284ce565518322bea32 Mon Sep 17 00:00:00 2001
From: MaheshRavishankar <mahesh.ravishankar@gmail.com>
Date: Thu, 22 Aug 2024 16:24:00 -0700
Subject: [PATCH] Fixes for LLVM change.
Signed-off-by: MaheshRavishankar <mahesh.ravishankar@gmail.com>
---
.../compiler/Codegen/Common/CPU/CPUPrepareUkernels.cpp | 8 +++++++-
1 file changed, 7 insertions(+), 1 deletion(-)
diff --git a/compiler/src/iree/compiler/Codegen/Common/CPU/CPUPrepareUkernels.cpp b/compiler/src/iree/compiler/Codegen/Common/CPU/CPUPrepareUkernels.cpp
index ca91e34af1349..0f8c38bebdd38 100644
--- a/compiler/src/iree/compiler/Codegen/Common/CPU/CPUPrepareUkernels.cpp
+++ b/compiler/src/iree/compiler/Codegen/Common/CPU/CPUPrepareUkernels.cpp
@@ -140,7 +140,13 @@ static LogicalResult reduceDefiningOp(PatternRewriter &rewriter, Value input) {
options.controlFn = [](Operation *op) -> SmallVector<unsigned> {
return {0};
};
- return linalg::dropUnitDims(rewriter, producer, options);
+ FailureOr<linalg::DropUnitDimsResult> result =
+ linalg::dropUnitDims(rewriter, producer, options);
+ if (failed(result)) {
+ return failure();
+ }
+ rewriter.replaceOp(producer, result->replacements);
+ return success();
}
/// Drops the first element from all the tile sizes list. The first element is