diff options
Diffstat (limited to 'include/DeserializationPasses.h')
-rw-r--r-- | include/DeserializationPasses.h | 9 |
1 files changed, 9 insertions, 0 deletions
diff --git a/include/DeserializationPasses.h b/include/DeserializationPasses.h index 1bc195a..1a38814 100644 --- a/include/DeserializationPasses.h +++ b/include/DeserializationPasses.h @@ -19,6 +19,8 @@ #include <memory> #include "mlir/Dialect/Func/IR/FuncOps.h" // from @llvm-project +#include "mlir/IR/BuiltinOps.h" // from @llvm-project +#include "mlir/IR/OwningOpRef.h" // from @llvm-project #include "mlir/Pass/Pass.h" // from @llvm-project namespace mlir { @@ -27,6 +29,13 @@ namespace tosa { std::unique_ptr<Pass> createTosaDeserializePass(); std::unique_ptr<Pass> createTosaDeserializeJSONPass(); +// deserializes a tosa file and return an mlir module +// if file_is_fbs is true, then treat file_name as a tosa flatbuffer file +// otherwise, treat file_name as a tosa json file +mlir::OwningOpRef<mlir::ModuleOp> +BuildMlirFromTosaFile(const char *file_name, mlir::MLIRContext *context, + bool file_is_fbs = true); + #define GEN_PASS_REGISTRATION #define GEN_PASS_CLASSES #include "include/DeserializationPasses.h.inc" |