diff options
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()"; } |