diff --git a/mlir/lib/Dialect/Tosa/Transforms/TosaMakeBroadcastable.cpp b/mlir/lib/Dialect/Tosa/Transforms/TosaMakeBroadcastable.cpp --- a/mlir/lib/Dialect/Tosa/Transforms/TosaMakeBroadcastable.cpp +++ b/mlir/lib/Dialect/Tosa/Transforms/TosaMakeBroadcastable.cpp @@ -269,6 +269,9 @@ RewritePatternSet patterns(func.getContext()); MLIRContext *ctx = func.getContext(); // Add the generated patterns to the list. + patterns.add>(ctx); + patterns.add>(ctx); + patterns.add>(ctx); patterns.add>(ctx); patterns.add>(ctx); patterns.add>(ctx); @@ -281,6 +284,10 @@ patterns.add>(ctx); patterns.add>(ctx); patterns.add>(ctx); + patterns.add>(ctx); + patterns.add>(ctx); + patterns.add>(ctx); + patterns.add>(ctx); (void)applyPatternsAndFoldGreedily(func, std::move(patterns)); } };