Commit 52e25523 authored by Nicolas Vasilache's avatar Nicolas Vasilache
Browse files

[mlir][Linalg] Fix incorrect erase order

parent d462aa5a

mlir/lib/Dialect/Linalg/Transforms/Hoisting.cpp

+4 −1
Original line number Diff line number Diff line
@@ -595,11 +595,14 @@ LogicalResult mlir::linalg::hoistPaddingOnTensors(SimplePadOp &simplePadOp, 
      b.create<SubTensorOp>(loc, simplePadOp.getResultType(), packedTensor,

                            offsets, sizes, strides)

          ->getResult(0));

  simplePadOp.erase();



  Operation *toErase = simplePadOp;



  // Make the newly cloned `simplePadOp` available to the caller.

  simplePadOp =

      cast<SimplePadOp>(bvm.lookup(simplePadOp.result()).getDefiningOp());



  toErase->erase();



  return success();

}