1. Mar 11, 2024
    • Matthias Springer's avatar
      [mlir][IR] Trigger `notifyOperationReplaced` on `replaceAllOpUsesWith` · 7b93ec38
      Matthias Springer authored
      Before this change: `notifyOperationReplaced` was triggered when calling `RewriteBase::replaceOp`.
      After this change: `notifyOperationReplaced` is triggered when `RewriterBase::replaceAllOpUsesWith` or `RewriterBase::replaceOp` is called.
      
      Until now, every `notifyOperationReplaced` was always sent together with a `notifyOperationErased`, which made that `notifyOperationErased` callback irrelevant. More importantly, when a user called `RewriterBase::replaceAllOpUsesWith`+`RewriterBase::eraseOp` instead of `RewriterBase::replaceOp`, no `notifyOperationReplaced` callback was sent, even though the two notations are semantically equivalent. As an example, this can be a problem when applying patterns with the transform dialect because the `TrackingListener` will only see the `notifyOperationErased` callback and the payload op is dropped from the mappings.
      
      Note: It is still possible to write semantically equivalent code that does not trigger a `notifyOperationReplaced` (e.g., when op results are replaced one-by-one), but this commit already improves the situation a lot.
      7b93ec38
  2. Mar 09, 2024