diff --git a/mlir/lib/Dialect/Transform/IR/TransformOps.cpp b/mlir/lib/Dialect/Transform/IR/TransformOps.cpp index 062606e7e10b6..415e2af491c07 100644 --- a/mlir/lib/Dialect/Transform/IR/TransformOps.cpp +++ b/mlir/lib/Dialect/Transform/IR/TransformOps.cpp @@ -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 argAttrsVec(argAttrs.begin(), argAttrs.end()); + state.getOrAddProperties().arg_attrs = + ArrayAttr::get(builder.getContext(), argAttrsVec); + } state.addRegion(); buildSequenceBody(builder, state, rootType,