diff options
| author | Jerry Ge <jerry.ge@arm.com> | 2022-09-19 21:07:38 -0700 |
|---|---|---|
| committer | Tai Ly <tai.ly@arm.com> | 2023-09-07 13:17:16 -0700 |
| commit | edc04e0ef1a93d1bf4578308d620dd49c10e1ad4 (patch) | |
| tree | 1044165c16528a17337f81aa15361c7b77aafe7f /include | |
| parent | 7566d1235cb646e46531c2eb34757cb4b3efa933 (diff) | |
| download | tosa_mlir_translator-edc04e0ef1a93d1bf4578308d620dd49c10e1ad4.tar.gz | |
[tosa_mlir_translator] Support Tosa StatefulOpsv0.90a0
Signed-off-by: Jerry Ge <jerry.ge@arm.com>
Change-Id: I6ce5a917cada436f6a80e6d85e670d6cd44e01e9
Diffstat (limited to 'include')
| -rw-r--r-- | include/SerializationPasses.h | 6 | ||||
| -rw-r--r-- | include/SerializationPasses.td | 2 |
2 files changed, 5 insertions, 3 deletions
diff --git a/include/SerializationPasses.h b/include/SerializationPasses.h index 66c6d80..c769b15 100644 --- a/include/SerializationPasses.h +++ b/include/SerializationPasses.h @@ -19,16 +19,18 @@ #include <memory> #include "mlir/Dialect/Func/IR/FuncOps.h" // from @llvm-project -#include "mlir/Pass/Pass.h" // from @llvm-project +#include "mlir/IR/BuiltinOps.h" +#include "mlir/Pass/Pass.h" // from @llvm-project namespace mlir { namespace tosa { -std::unique_ptr<Pass> createTosaSerializePass(); +std::unique_ptr<OperationPass<ModuleOp>> createTosaSerializePass(); std::unique_ptr<Pass> createTosaSerializeJSONPass(); #define GEN_PASS_REGISTRATION #define GEN_PASS_CLASSES +#define GEN_PASS_DECL_TOSASERIALIZEPASS #include "include/SerializationPasses.h.inc" } // namespace tosa diff --git a/include/SerializationPasses.td b/include/SerializationPasses.td index 3bdeb1b..9cfc204 100644 --- a/include/SerializationPasses.td +++ b/include/SerializationPasses.td @@ -14,7 +14,7 @@ include "mlir/Pass/PassBase.td" -def TosaSerializationPass : Pass<"tosa-serialize", "func::FuncOp"> { +def TosaSerializationPass : Pass<"tosa-serialize", "mlir::ModuleOp"> { let summary = "Generate TOSA flatbuffer serialized form"; let constructor = "createTosaSerializePass()"; } |