diff --git a/include/scalehls/Dialect/HLS/IR/HLSUIPOps.td b/include/scalehls/Dialect/HLS/IR/HLSUIPOps.td index e11f1fec..5d9ab10c 100644 --- a/include/scalehls/Dialect/HLS/IR/HLSUIPOps.td +++ b/include/scalehls/Dialect/HLS/IR/HLSUIPOps.td @@ -65,17 +65,17 @@ def InstanceOp : HLSOp<"uip.instance", [ }]; } -def PortOp : HLSOp<"uip.port", [Symbol, HasParent<"DeclareOp">]> { +def PortOp : HLSOp<"uip.port", [AttrSizedOperandSegments, Symbol, HasParent<"DeclareOp">]> { let summary = "Declare a port of an IP"; let arguments = (ins TypeType:$type, Variadic:$sizes, - MemRefLayoutAttrInterface:$layout, PortKindAttr:$kind, + MemRefLayoutAttrInterface:$layout, PortKindAttr:$kind, Variadic:$symbol_ref, OptionalAttr:$value, SymbolNameAttr:$sym_name); let results = (outs PortType:$result); let assemblyFormat = [{ - $sym_name $kind `type` $type `sizes` `(` $sizes `)` $layout attr-dict - `:` functional-type($sizes, $result) + $sym_name $kind `type` $type `sizes` `(` $sizes `)` `symbol_ref` `(` $symbol_ref `)` `(` type($symbol_ref) `)` $layout attr-dict + `:` functional-type($sizes, $result) }]; } diff --git a/test/EmitHLSCpp/test-instanceOp.mlir b/test/EmitHLSCpp/test-instanceOp.mlir index 2945d67b..147fd3cc 100644 --- a/test/EmitHLSCpp/test-instanceOp.mlir +++ b/test/EmitHLSCpp/test-instanceOp.mlir @@ -11,11 +11,11 @@ module attributes { torch.debug_module_name = "MLP" } { %1 = hls.dse.param @template1