| //===- MathDialect.cpp - MLIR dialect for Math implementation -------------===// |
| // |
| // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. |
| // See https://llvm.org/LICENSE.txt for license information. |
| // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception |
| // |
| //===----------------------------------------------------------------------===// |
| |
| #include "mlir/Dialect/Math/IR/Math.h" |
| #include "mlir/Transforms/InliningUtils.h" |
| |
| using namespace mlir; |
| using namespace mlir::math; |
| |
| #include "mlir/Dialect/Math/IR/MathOpsDialect.cpp.inc" |
| |
| namespace { |
| /// This class defines the interface for handling inlining with math |
| /// operations. |
| struct MathInlinerInterface : public DialectInlinerInterface { |
| using DialectInlinerInterface::DialectInlinerInterface; |
| |
| /// All operations within math ops can be inlined. |
| bool isLegalToInline(Operation *, Region *, bool, |
| BlockAndValueMapping &) const final { |
| return true; |
| } |
| }; |
| } // end anonymous namespace |
| |
| void mlir::math::MathDialect::initialize() { |
| addOperations< |
| #define GET_OP_LIST |
| #include "mlir/Dialect/Math/IR/MathOps.cpp.inc" |
| >(); |
| addInterfaces<MathInlinerInterface>(); |
| } |