[mlir][Transform] Fix ASAN heap-use-after-free

diff --git a/mlir/lib/Dialect/Transform/IR/TransformInterfaces.cpp b/mlir/lib/Dialect/Transform/IR/TransformInterfaces.cpp
index 9b85af3..5d84b7b 100644 (file)
--- a/mlir/lib/Dialect/Transform/IR/TransformInterfaces.cpp
+++ b/mlir/lib/Dialect/Transform/IR/TransformInterfaces.cpp
@@ -130,7 +130,7 @@ LogicalResult transform::TransformState::updatePayloadOps(
   if (failed(result.checkAndReport()))
     return failure();
 
-  std::swap(association, updated);
+  it->second = updated;
   return success();
 }