diff options
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 |