[Mlir-commits] [mlir] cf2d80b - [mlir][Transform] Fix transform::SequenceOp builder

[Nicolas Vasilache]

Author: Nicolas Vasilache
Date: 2023-03-02T01:11:08-08:00
New Revision: cf2d80b9a2a4ff6b53ca3c9efd5c740cd733e42d

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

LOG: [mlir][Transform] Fix transform::SequenceOp builder

Previously, the wrond type would be used and would not matche the number of locations triggering
an assertion failures in `createBlock`.

Added: 
    

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

Removed: 
    


################################################################################
diff  --git a/mlir/lib/Dialect/Transform/IR/TransformOps.cpp b/mlir/lib/Dialect/Transform/IR/TransformOps.cpp
index b207810edb5a..21b3b2dd2045 100644
--- a/mlir/lib/Dialect/Transform/IR/TransformOps.cpp
+++ b/mlir/lib/Dialect/Transform/IR/TransformOps.cpp
@@ -973,8 +973,9 @@ static void buildSequenceBody(OpBuilder &builder, OperationState &state,
 
   OpBuilder::InsertionGuard guard(builder);
   Region *region = state.regions.back().get();
-  Block *bodyBlock = builder.createBlock(region, region->begin(),
-                                         extraBindingTypes, {state.location});
+  Block *bodyBlock =
+      builder.createBlock(region, region->begin(), types,
+                          SmallVector<Location>(types.size(), state.location));
 
   // Populate body.
   builder.setInsertionPointToStart(bodyBlock);