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/SerializationPasses.h | |
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/SerializationPasses.h')
-rw-r--r-- | include/SerializationPasses.h | 6 |
1 files changed, 4 insertions, 2 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 |