diff options
Diffstat (limited to 'include')
-rw-r--r-- | include/SerializationPasses.h | 3 | ||||
-rw-r--r-- | include/SerializationPasses.td | 4 |
2 files changed, 4 insertions, 3 deletions
diff --git a/include/SerializationPasses.h b/include/SerializationPasses.h index 2c04ca1..66c6d80 100644 --- a/include/SerializationPasses.h +++ b/include/SerializationPasses.h @@ -18,7 +18,8 @@ #include <memory> -#include "mlir/Pass/Pass.h" // from @llvm-project +#include "mlir/Dialect/Func/IR/FuncOps.h" // from @llvm-project +#include "mlir/Pass/Pass.h" // from @llvm-project namespace mlir { namespace tosa { diff --git a/include/SerializationPasses.td b/include/SerializationPasses.td index ec272e4..3bdeb1b 100644 --- a/include/SerializationPasses.td +++ b/include/SerializationPasses.td @@ -14,12 +14,12 @@ include "mlir/Pass/PassBase.td" -def TosaSerializationPass : Pass<"tosa-serialize", "FuncOp"> { +def TosaSerializationPass : Pass<"tosa-serialize", "func::FuncOp"> { let summary = "Generate TOSA flatbuffer serialized form"; let constructor = "createTosaSerializePass()"; } -def TosaSerializationJSONPass : Pass<"tosa-serialize-json", "FuncOp"> { +def TosaSerializationJSONPass : Pass<"tosa-serialize-json", "func::FuncOp"> { let summary = "Generate TOSA flatbuffer JSON form"; let constructor = "createTosaSerializeJSONPass()"; } |