[Mlir-commits] [mlir] 55292e9 - [mlir][Transform] Fix ASAN heap-use-after-free

[Nicolas Vasilache]

Author: Nicolas Vasilache
Date: 2022-11-02T09:47:14-07:00
New Revision: 55292e9ee152e4421b5e52e6a9247d479abdb0e1

URL: https://github.com/llvm/llvm-project/commit/55292e9ee152e4421b5e52e6a9247d479abdb0e1
DIFF: https://github.com/llvm/llvm-project/commit/55292e9ee152e4421b5e52e6a9247d479abdb0e1.diff

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

Added: 
    

Modified: 
    mlir/lib/Dialect/Transform/IR/TransformInterfaces.cpp

Removed: 
    


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