}
namespace {
-struct ConvertStdElementwiseOpOnRankedTensors : public RewritePattern {
- ConvertStdElementwiseOpOnRankedTensors()
+struct ConvertAnyElementwiseMappableOpOnRankedTensors : public RewritePattern {
+ ConvertAnyElementwiseMappableOpOnRankedTensors()
: RewritePattern(/*benefit=*/1, MatchAnyOpTypeTag()) {}
LogicalResult matchAndRewrite(Operation *op,
PatternRewriter &rewriter) const final {
void mlir::populateElementwiseToLinalgConversionPatterns(
OwningRewritePatternList &patterns, MLIRContext *) {
- patterns.insert<ConvertStdElementwiseOpOnRankedTensors>();
+ patterns.insert<ConvertAnyElementwiseMappableOpOnRankedTensors>();
}
namespace {