From 55292e9ee152e4421b5e52e6a9247d479abdb0e1 Mon Sep 17 00:00:00 2001
From: Nicolas Vasilache <nicolas.vasilache@gmail.com>
Date: Wed, 2 Nov 2022 09:39:16 -0700
Subject: [PATCH] [mlir][Transform] Fix ASAN heap-use-after-free

---
 mlir/lib/Dialect/Transform/IR/TransformInterfaces.cpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/mlir/lib/Dialect/Transform/IR/TransformInterfaces.cpp b/mlir/lib/Dialect/Transform/IR/TransformInterfaces.cpp
index 9b85af3..5d84b7b 100644
--- 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();
 }
 
-- 
2.7.4