Skip to content
Open
5 changes: 5 additions & 0 deletions mlir/lib/Dialect/Transform/IR/TransformOps.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -2571,6 +2571,11 @@ void transform::NamedSequenceOp::build(OpBuilder &builder,
TypeAttr::get(FunctionType::get(builder.getContext(),
rootType, resultTypes)));
state.attributes.append(attrs.begin(), attrs.end());
if (!argAttrs.empty()) {
SmallVector<Attribute> argAttrsVec(argAttrs.begin(), argAttrs.end());
state.getOrAddProperties<Properties>().arg_attrs =
ArrayAttr::get(builder.getContext(), argAttrsVec);
}
state.addRegion();

buildSequenceBody(builder, state, rootType,
Expand Down