diff options
-rw-r--r-- | mlir/lib/Dialect/Async/Transforms/AsyncRuntimeRefCounting.cpp | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/mlir/lib/Dialect/Async/Transforms/AsyncRuntimeRefCounting.cpp b/mlir/lib/Dialect/Async/Transforms/AsyncRuntimeRefCounting.cpp index ddc64ea7ebf9..91e37dd9ac36 100644 --- a/mlir/lib/Dialect/Async/Transforms/AsyncRuntimeRefCounting.cpp +++ b/mlir/lib/Dialect/Async/Transforms/AsyncRuntimeRefCounting.cpp @@ -248,7 +248,7 @@ LogicalResult AsyncRuntimeRefCountingPass::addDropRefAfterLastUse(Value value) { Region *definingRegion = value.getParentRegion(); // Last users of the `value` inside all blocks where the value dies. - llvm::SmallSet<Operation *, 4> lastUsers; + llvm::SmallPtrSet<Operation *, 4> lastUsers; // Find blocks in the `definingRegion` that have users of the `value` (if // there are multiple users in the block, which one will be selected is |