From 79d918e2d54ebe0f14262d3e7c545c6aa1d4ea6d Mon Sep 17 00:00:00 2001 From: Tina Jung Date: Thu, 27 Mar 2025 14:21:53 +0000 Subject: [PATCH] Check valid emitc float/opaque types, not float Use the emitc-provided function to check the types instead of checking for float types, as the other arith lowering do. --- mlir/lib/Conversion/ArithToEmitC/ArithToEmitC.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mlir/lib/Conversion/ArithToEmitC/ArithToEmitC.cpp b/mlir/lib/Conversion/ArithToEmitC/ArithToEmitC.cpp index 211f00807d72b..af7e0fa90a419 100644 --- a/mlir/lib/Conversion/ArithToEmitC/ArithToEmitC.cpp +++ b/mlir/lib/Conversion/ArithToEmitC/ArithToEmitC.cpp @@ -84,7 +84,7 @@ class CmpFOpConversion : public OpConversionPattern { matchAndRewrite(arith::CmpFOp op, OpAdaptor adaptor, ConversionPatternRewriter &rewriter) const override { - if (!isa(adaptor.getRhs().getType())) { + if (!emitc::isFloatOrOpaqueType(adaptor.getRhs().getType())) { return rewriter.notifyMatchFailure(op.getLoc(), "cmpf currently only supported on " "floats, not tensors/vectors thereof");