[Mlir-commits] [mlir] [mlir][transform] Emit error message with `emitSilenceableFailure` (PR #86146)
Oleksandr Alex Zinenko
llvmlistbot at llvm.org
Fri Mar 22 03:59:32 PDT 2024
================
@@ -3269,22 +3269,24 @@ DiagnosedSilenceableFailure transform::FlattenElementwiseLinalgOp::applyToOne(
transform::ApplyToEachResultList &results,
transform::TransformState &state) {
rewriter.setInsertionPoint(target);
- if (!isElementwise(target)) {
- failed(rewriter.notifyMatchFailure(
- target, "only elementwise flattening is supported"));
- return emitDefaultSilenceableFailure(target);
- }
+ if (!isElementwise(target))
+ return mlir::emitSilenceableFailure(target->getLoc())
+ << "only elementwise flattening is supported";
+
// If rank <= 1, do nothing
if (target.getNumLoops() <= 1) {
results.push_back(target);
return DiagnosedSilenceableFailure::success();
}
+
+ // Attempt to flatten all dims to one
ReassociationIndices reassociation(target.getNumLoops());
std::iota(reassociation.begin(), reassociation.end(), 0);
auto maybeFlattened =
collapseOpIterationDims(target, reassociation, rewriter);
if (failed(maybeFlattened))
- return emitDefaultSilenceableFailure(target);
+ return mlir::emitSilenceableFailure(target->getLoc())
+ << "Attempted to flatten, but failed";
----------------
ftynse wrote:
Nit: error messages should start with a lowercase letter.
https://github.com/llvm/llvm-project/pull/86146
More information about the Mlir-commits
mailing list