commit 826f1db1f6ddf6210c857781492ebc03fd856327
author Stella Laurenzo <laurenzo@google.com> Fri Nov 19 12:54:23 2021 -0800
committer Copybara-Service <iree-copybara-bot@google.com> Fri Nov 19 12:55:32 2021 -0800
tree b1684b84565834ff4095ab3c8ce85774f15d6711
parent 17deac0a0de1e2c8b8ae4435fd65d3743275b3dc

Align FileCheck options with upstream defaults.

PiperOrigin-RevId: 411124255

llvm-external-projects/iree-dialects/lib/Dialect/PyDM/IR/PyDMOps.cpp

diff --git a/llvm-external-projects/iree-dialects/lib/Dialect/PyDM/IR/PyDMOps.cpp b/llvm-external-projects/iree-dialects/lib/Dialect/PyDM/IR/PyDMOps.cpp
index 7ef7bfc..3b3e961 100644
--- a/llvm-external-projects/iree-dialects/lib/Dialect/PyDM/IR/PyDMOps.cpp
+++ b/llvm-external-projects/iree-dialects/lib/Dialect/PyDM/IR/PyDMOps.cpp
@@ -697,12 +697,10 @@
 // RaiseOnFailureOp
 //===----------------------------------------------------------------------===//
 
-LogicalResult PYDM::RaiseOnFailureOp::fold(
-    ArrayRef<Attribute> operands, SmallVectorImpl<OpFoldResult> &results) {
-  assert(operands.size() == 1 && "expected one fold operand");
-  // Unit exception result is success. Just elide.
-  if (operands[0] && operands[0].isa<UnitAttr>()) {
-    erase();
+LogicalResult PYDM::RaiseOnFailureOp::canonicalize(RaiseOnFailureOp op,
+                                                   PatternRewriter &rewriter) {
+  if (op.exc_result().getDefiningOp<SuccessOp>()) {
+    op.getOperation()->erase();
     return success();
   }
   return failure();