18 #define GEN_PASS_DEF_CONVERTCOMPLEXTOLIBM
19 #include "mlir/Conversion/Passes.h.inc"
27 struct ComplexTypeResolver {
28 std::optional<bool> operator()(
Type type)
const {
29 auto complexType = cast<ComplexType>(type);
30 auto elementType = complexType.getElementType();
31 if (!isa<Float32Type, Float64Type>(elementType))
34 return elementType.getIntOrFloatBitWidth() == 64;
40 struct FloatTypeResolver {
41 std::optional<bool> operator()(
Type type)
const {
42 auto elementType = cast<FloatType>(type);
43 if (!isa<Float32Type, Float64Type>(elementType))
46 return elementType.getIntOrFloatBitWidth() == 64;
54 template <
typename Op,
typename TypeResolver = ComplexTypeResolver>
58 ScalarOpToLibmCall(
MLIRContext *context, StringRef floatFunc,
61 doubleFunc(doubleFunc){};
66 std::string floatFunc, doubleFunc;
70 template <
typename Op,
typename TypeResolver>
71 LogicalResult ScalarOpToLibmCall<Op, TypeResolver>::matchAndRewrite(
74 auto isDouble = TypeResolver()(op.getType());
75 if (!isDouble.has_value())
78 auto name = *isDouble ? doubleFunc : floatFunc;
80 auto opFunc = dyn_cast_or_null<SymbolOpInterface>(
87 rewriter.
getContext(), op->getOperandTypes(), op->getResultTypes());
102 patterns.
add<ScalarOpToLibmCall<complex::PowOp>>(patterns.
getContext(),
103 "cpowf",
"cpow", benefit);
104 patterns.
add<ScalarOpToLibmCall<complex::SqrtOp>>(patterns.
getContext(),
105 "csqrtf",
"csqrt", benefit);
106 patterns.
add<ScalarOpToLibmCall<complex::TanhOp>>(patterns.
getContext(),
107 "ctanhf",
"ctanh", benefit);
108 patterns.
add<ScalarOpToLibmCall<complex::CosOp>>(patterns.
getContext(),
109 "ccosf",
"ccos", benefit);
110 patterns.
add<ScalarOpToLibmCall<complex::SinOp>>(patterns.
getContext(),
111 "csinf",
"csin", benefit);
112 patterns.
add<ScalarOpToLibmCall<complex::ConjOp>>(patterns.
getContext(),
113 "conjf",
"conj", benefit);
114 patterns.
add<ScalarOpToLibmCall<complex::LogOp>>(patterns.
getContext(),
115 "clogf",
"clog", benefit);
116 patterns.
add<ScalarOpToLibmCall<complex::AbsOp, FloatTypeResolver>>(
117 patterns.
getContext(),
"cabsf",
"cabs", benefit);
118 patterns.
add<ScalarOpToLibmCall<complex::AngleOp, FloatTypeResolver>>(
119 patterns.
getContext(),
"cargf",
"carg", benefit);
120 patterns.
add<ScalarOpToLibmCall<complex::TanOp>>(patterns.
getContext(),
121 "ctanf",
"ctan", benefit);
125 struct ConvertComplexToLibmPass
126 :
public impl::ConvertComplexToLibmBase<ConvertComplexToLibmPass> {
127 void runOnOperation()
override;
131 void ConvertComplexToLibmPass::runOnOperation() {
132 auto module = getOperation();
138 target.addLegalDialect<func::FuncDialect>();
139 target.addIllegalOp<complex::PowOp, complex::SqrtOp, complex::TanhOp,
140 complex::CosOp, complex::SinOp, complex::ConjOp,
141 complex::LogOp, complex::AbsOp, complex::AngleOp,
147 std::unique_ptr<OperationPass<ModuleOp>>
149 return std::make_unique<ConvertComplexToLibmPass>();
static MLIRContext * getContext(OpFoldResult val)
MLIRContext * getContext() const
This class describes a specific conversion target.
MLIRContext is the top-level object for a collection of MLIR operations.
RAII guard to reset the insertion point of the builder when destroyed.
void setInsertionPointToStart(Block *block)
Sets the insertion point to the start of the specified block.
Operation * create(const OperationState &state)
Creates an operation given the fields represented as an OperationState.
This provides public APIs that all operations should have.
This class represents the benefit of a pattern match in a unitless scheme that ranges from 0 (very li...
A special type of RewriterBase that coordinates the application of a rewrite pattern on the current I...
MLIRContext * getContext() const
RewritePatternSet & add(ConstructorArg &&arg, ConstructorArgs &&...args)
Add an instance of each of the pattern types 'Ts' to the pattern list with the given arguments.
OpTy replaceOpWithNewOp(Operation *op, Args &&...args)
Replace the results of the given (original) op with a new op that is created without verification (re...
static Operation * lookupSymbolIn(Operation *op, StringAttr symbol)
Returns the operation registered with the given symbol name with the regions of 'symbolTableOp'.
static Operation * getNearestSymbolTable(Operation *from)
Returns the nearest symbol table from a given operation from.
Instances of the Type class are uniqued, have an immutable identifier and an optional mutable compone...
Include the generated interface declarations.
std::unique_ptr< OperationPass< ModuleOp > > createConvertComplexToLibmPass()
Create a pass to convert Complex operations to libm calls.
auto get(MLIRContext *context, Ts &&...params)
Helper method that injects context only if needed, this helps unify some of the attribute constructio...
LogicalResult applyPartialConversion(ArrayRef< Operation * > ops, const ConversionTarget &target, const FrozenRewritePatternSet &patterns, ConversionConfig config=ConversionConfig())
Below we define several entry points for operation conversion.
void populateComplexToLibmConversionPatterns(RewritePatternSet &patterns, PatternBenefit benefit)
Populate the given list with patterns that convert from Complex to Libm calls.
OpRewritePattern is a wrapper around RewritePattern that allows for matching and rewriting against an...