Specialize.cpp File Reference
#include "mlir/Analysis/SliceAnalysis.h"#include "mlir/Dialect/Complex/IR/Complex.h"#include "mlir/Dialect/Linalg/IR/Linalg.h"#include "mlir/Dialect/Linalg/IR/LinalgInterfaces.h"#include "mlir/Dialect/Linalg/Passes.h"#include "mlir/Dialect/Linalg/Transforms/Transforms.h"#include "mlir/Dialect/Math/IR/Math.h"#include "mlir/IR/PatternMatch.h"#include "mlir/Transforms/GreedyPatternRewriteDriver.h"#include "mlir/Dialect/Linalg/Passes.h.inc"Go to the source code of this file.
Namespaces | |
| namespace | mlir |
| Include the generated interface declarations. | |
Macros | |
| #define | GEN_PASS_DEF_LINALGSPECIALIZEGENERICOPSPASS |
| #define | DEBUG_TYPE "linalg-specialization" |
| #define | REPLACE_BINARY_OP(NEWOP, OPERANDS_SWAP) |
| #define | REPLACE_UNARY_OP(NEWOP) |
| #define | CONV_OP_SPECIALIZER(ConvOpTy) |
Functions | |
| static bool | areBinOpsSwapped (GenericOp genericOp) |
Macro Definition Documentation
◆ CONV_OP_SPECIALIZER
| #define CONV_OP_SPECIALIZER | ( | ConvOpTy | ) |
Value:
if (std::optional<DilationsAndStrides> convParams = \
matchConvolutionOpOfType<ConvOpTy>(genericOp)) \
return specializeToConvOp<ConvOpTy>( \
rewriter, genericOp, convParams->dilations, convParams->strides); \
std::optional< DilationsAndStrides > matchConvolutionOpOfType(LinalgOp op)
Given a linalg op this function returns DilationsAndStrides if it is a convolution op of type ConvOpT...
◆ DEBUG_TYPE
| #define DEBUG_TYPE "linalg-specialization" |
Definition at line 29 of file Specialize.cpp.
◆ GEN_PASS_DEF_LINALGSPECIALIZEGENERICOPSPASS
| #define GEN_PASS_DEF_LINALGSPECIALIZEGENERICOPSPASS |
Definition at line 25 of file Specialize.cpp.
◆ REPLACE_BINARY_OP
| #define REPLACE_BINARY_OP | ( | NEWOP, | |
| OPERANDS_SWAP ) |
Value:
(rewriter.replaceOpWithNewOp<NEWOP>( \
genericOp, \
ValueRange{genericOp.getDpsInputs()[(OPERANDS_SWAP) ? 1 : 0], \
genericOp.getDpsInputs()[(OPERANDS_SWAP) ? 0 : 1]}, \
ValueRange{genericOp.getDpsInits()[0]}))
This class provides an abstraction over the different types of ranges over Values.
Definition ValueRange.h:387
Definition at line 31 of file Specialize.cpp.
Referenced by mlir::linalg::specializeGenericOp().
◆ REPLACE_UNARY_OP
| #define REPLACE_UNARY_OP | ( | NEWOP | ) |
Value:
(rewriter.replaceOpWithNewOp<NEWOP>(genericOp, \
ValueRange{genericOp.getDpsInputs()[0]}, \
ValueRange{genericOp.getDpsInits()[0]}))
Definition at line 38 of file Specialize.cpp.
Referenced by mlir::linalg::specializeGenericOp().
Function Documentation
◆ areBinOpsSwapped()
|
static |
Definition at line 57 of file Specialize.cpp.
References mlir::Block::front(), mlir::IROperand< DerivedT, IRValueT >::get(), mlir::Block::getArgument(), and mlir::Operation::getOpOperand().
Referenced by mlir::linalg::specializeGenericOp().
Generated on for MLIR by
1.14.0