1 files changed, 4 insertions, 2 deletions
diff --git a/include/SerializationPasses.h b/include/SerializationPasses.h
index 0991f87..2c04ca1 100644
--- a/include/SerializationPasses.h
+++ b/include/SerializationPasses.h
@@ -23,10 +23,12 @@
 namespace mlir {
 namespace tosa {
 
-std::unique_ptr<OperationPass<FuncOp>> createTosaSerializePass();
+std::unique_ptr<Pass> createTosaSerializePass();
+std::unique_ptr<Pass> createTosaSerializeJSONPass();
 
 #define GEN_PASS_REGISTRATION
-#include "SerializationPasses.h.inc"
+#define GEN_PASS_CLASSES
+#include "include/SerializationPasses.h.inc"
 
 } // namespace tosa
 } // namespace mlir