diff options
Diffstat (limited to 'include')
-rw-r--r-- | include/SerializationPasses.h | 6 | ||||
-rw-r--r-- | include/SerializationPasses.td | 8 |
2 files changed, 10 insertions, 4 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 diff --git a/include/SerializationPasses.td b/include/SerializationPasses.td index 6df996e..ec272e4 100644 --- a/include/SerializationPasses.td +++ b/include/SerializationPasses.td @@ -1,7 +1,7 @@ // Copyright (c) 2020-2021, ARM Limited. // -// Licensed under the Apache License, Version 2.0 with LLVM Exceptions -// (the "License"); you may not use this file except in compliance with +// Licensed under the Apache License, Version 2.0 with LLVM Exceptions +// (the "License"); you may not use this file except in compliance with // the License. You may obtain a copy of the License at // // https://llvm.org/LICENSE.txt @@ -19,3 +19,7 @@ def TosaSerializationPass : Pass<"tosa-serialize", "FuncOp"> { let constructor = "createTosaSerializePass()"; } +def TosaSerializationJSONPass : Pass<"tosa-serialize-json", "FuncOp"> { + let summary = "Generate TOSA flatbuffer JSON form"; + let constructor = "createTosaSerializeJSONPass()"; +} |