[mlir][SCF] Make the scf.take_assumed_branch transform only read its target handle
authorNicolas Vasilache <nicolas.vasilache@gmail.com>
Wed, 12 Apr 2023 19:12:46 +0000 (12:12 -0700)
committerNicolas Vasilache <nicolas.vasilache@gmail.com>
Wed, 12 Apr 2023 19:30:18 +0000 (12:30 -0700)
commit5b66e44cff1ff19fc72a264b00e8b83250c86863
treeea0b21d4d872e8b41cbc37a41694b65df8f30df7
parent3c83aeee6b7d5e9ac5de1e29605ee57c2b800ca1
[mlir][SCF] Make the scf.take_assumed_branch transform only read its target handle

Handles are tracked properly and usage composes better if we don't artificially lose handles.

Differential Revision: https://reviews.llvm.org/D148155
mlir/lib/Dialect/SCF/TransformOps/SCFTransformOps.cpp
mlir/test/Dialect/SCF/transform-op-take-assumed-branch.mlir