aboutsummaryrefslogtreecommitdiff
path: root/include/SerializationPasses.h
diff options
context:
space:
mode:
authorJerry Ge <jerry.ge@arm.com>2022-09-19 21:07:38 -0700
committerTai Ly <tai.ly@arm.com>2023-09-07 13:17:16 -0700
commitedc04e0ef1a93d1bf4578308d620dd49c10e1ad4 (patch)
tree1044165c16528a17337f81aa15361c7b77aafe7f /include/SerializationPasses.h
parent7566d1235cb646e46531c2eb34757cb4b3efa933 (diff)
downloadtosa_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.h6
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