-
Notifications
You must be signed in to change notification settings - Fork 4
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Signed-off-by: Anton Korobeynikov <anton@korobeynikov.info>
- Loading branch information
Showing
2 changed files
with
165 additions
and
159 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,164 @@ | ||
| #ifndef P4MLIR_DIALECT_P4HIR_P4HIR_PARSEROPS_TD | ||
| #define P4MLIR_DIALECT_P4HIR_P4HIR_PARSEROPS_TD | ||
|
|
||
| def ParserOp : P4HIR_Op<"parser", | ||
| [Symbol, SymbolTable, IsolatedFromAbove, | ||
| FunctionOpInterface, AutomaticAllocationScope]> { | ||
| let arguments = (ins SymbolNameAttr:$sym_name, | ||
| TypeAttrOf<FuncType>:$applyType, | ||
| TypeAttrOf<FuncType>:$constructorType); | ||
| let regions = (region SizedRegion<1>:$body); | ||
| let hasCustomAssemblyFormat = 1; | ||
|
|
||
| let extraClassDeclaration = [{ | ||
| mlir::Region *getCallableRegion() { return &getBody(); } | ||
|
|
||
| auto getFunctionType() { return getApplyType(); } | ||
|
|
||
| llvm::ArrayRef<mlir::Type> getArgumentTypes() { | ||
| return getApplyType().getInputs(); | ||
| } | ||
|
|
||
| void setFunctionTypeAttr(mlir::TypeAttr attr) { | ||
| getProperties().applyType = attr; | ||
| } | ||
|
|
||
| llvm::ArrayRef<mlir::Type> getResultTypes() { | ||
| return {}; | ||
| } | ||
| }]; | ||
| } | ||
|
|
||
| def ParserStateOp : P4HIR_Op<"state", | ||
| [Symbol, AutomaticAllocationScope]> { | ||
| let arguments = (ins SymbolNameAttr:$sym_name); | ||
| let regions = (region MinSizedRegion<1>:$body); | ||
| let assemblyFormat = [{ | ||
| $sym_name $body attr-dict | ||
| }]; | ||
| } | ||
|
|
||
| def ParserTransitionOp : P4HIR_Op<"transition", | ||
| [Terminator, | ||
| ParentOneOf<["ParserStateOp", "ParserOp"]>, | ||
| DeclareOpInterfaceMethods<SymbolUserOpInterface>]> { | ||
| let arguments = (ins SymbolRefAttr:$state); | ||
| let assemblyFormat = [{ | ||
| `to` $state attr-dict | ||
| }]; | ||
| } | ||
|
|
||
| def ParserAcceptOp : P4HIR_Op<"parser_accept", | ||
| [Terminator, | ||
| ParentOneOf<["ParserStateOp"]>]> { | ||
| let assemblyFormat = "attr-dict"; | ||
| } | ||
|
|
||
| def ParserRejectOp : P4HIR_Op<"parser_reject", | ||
| [Terminator, | ||
| ParentOneOf<["ParserStateOp", | ||
| "IfOp" // for verify() lowering | ||
| ]>]> { | ||
| let arguments = (ins OptionalAttr<P4HIR_ErrorCodeAttr>:$error); | ||
| let assemblyFormat = "attr-dict (`with` `error` $error^)?"; | ||
|
|
||
| let builders = [ | ||
| // Convenience builder without an error | ||
| OpBuilder<(ins), [{ | ||
| build($_builder, $_state, {}); | ||
| }]> | ||
| ]; | ||
| } | ||
|
|
||
| def ParserTransitionSelectOp : P4HIR_Op<"transition_select", | ||
| [Terminator, NoTerminator, | ||
| ParentOneOf<["ParserStateOp"]>]> { | ||
| // FIXME: constraint argument better. Should be tuple or "normal" type. | ||
| let arguments = (ins AnyP4Type:$select); | ||
| let regions = (region SizedRegion<1>:$body); | ||
| let assemblyFormat = [{ | ||
| $select `:` type($select) $body attr-dict | ||
| }]; | ||
| // FIXME: Implement verifier for cases | ||
| } | ||
|
|
||
| def ParserSelectCaseOp : P4HIR_Op<"select_case", | ||
| [AutomaticAllocationScope, | ||
| ParentOneOf<["ParserTransitionSelectOp"]>, | ||
| DeclareOpInterfaceMethods<SymbolUserOpInterface>]> { | ||
| let arguments = (ins SymbolRefAttr:$state); | ||
| let regions = (region MinSizedRegion<1>:$key); | ||
| let assemblyFormat = [{ | ||
| $key `to` $state attr-dict | ||
| }]; | ||
| // TBD: check region | ||
| let hasVerifier = 0; | ||
|
|
||
| let skipDefaultBuilders = 1; | ||
| let builders = [ | ||
| OpBuilder<(ins "llvm::function_ref<void(mlir::OpBuilder &, mlir::Location)>":$keyBuilder, | ||
| "llvm::StringRef":$nextState)>, | ||
| OpBuilder<(ins "llvm::function_ref<void(mlir::OpBuilder &, mlir::Location)>":$keyBuilder, | ||
| "mlir::SymbolRefAttr":$nextState)> | ||
| ]; | ||
| } | ||
|
|
||
| def SetOp : P4HIR_Op<"set", | ||
| [Pure, SameTypeOperands, | ||
| DeclareOpInterfaceMethods<OpAsmOpInterface, ["getAsmResultNames"]>]> { | ||
| let summary = "Create a set from constituent parts."; | ||
| // FIXME: Better constraint type | ||
| let arguments = (ins Variadic<AnyP4Type>:$input); | ||
| let results = (outs SetType:$result); | ||
| let hasCustomAssemblyFormat = 1; | ||
| // FIXME: use declarative format | ||
| // let assemblyFormat = [{ | ||
| // `(` $input `)` attr-dict `:` type($result) | ||
| // }]; | ||
|
|
||
| // TODO: Automatically infer result type | ||
| let builders = [ | ||
| OpBuilder<(ins "mlir::ValueRange":$input)> | ||
| ]; | ||
|
|
||
| let hasVerifier = 1; | ||
| } | ||
|
|
||
| def SetProductOp : P4HIR_Op<"set_product", | ||
| [Pure, | ||
| DeclareOpInterfaceMethods<OpAsmOpInterface, ["getAsmResultNames"]>]> { | ||
| let summary = "Create a carthesian set product"; | ||
| // FIXME: Better constraint type | ||
| let arguments = (ins Variadic<SetType>:$input); | ||
| let results = (outs SetType:$result); | ||
| let hasCustomAssemblyFormat = 1; | ||
| // FIXME: use declarative format | ||
| // let assemblyFormat = [{ | ||
| // `(` $input `)` attr-dict `:` type($result) | ||
| // }]; | ||
|
|
||
| // TODO: Automatically infer result type | ||
| let builders = [ | ||
| OpBuilder<(ins "mlir::ValueRange":$input)> | ||
| ]; | ||
|
|
||
| let hasVerifier = 1; | ||
| } | ||
|
|
||
| def UniversalSetOp : P4HIR_Op<"universal_set", | ||
| [Pure, | ||
| DeclareOpInterfaceMethods<OpAsmOpInterface, ["getAsmResultNames"]>]> { | ||
| let summary = "Create a universal set"; | ||
| // TODO: Make it properly types | ||
| // let arguments = (ins AnyP4Type:$input); | ||
| let results = (outs SetType:$result); | ||
| let assemblyFormat = [{ | ||
| attr-dict `:` qualified(type($result)) | ||
| }]; | ||
|
|
||
| let builders = [ | ||
| OpBuilder<(ins)> | ||
| ]; | ||
| } | ||
|
|
||
| #endif // P4MLIR_DIALECT_P4HIR_P4HIR_PARSEROPS_TD |