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 "llvm/ADT/TypeSwitch.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	CONV_OP_SPECIALIZER(ConvOpTy)

Functions
static bool	areBinOpsSwapped (GenericOp genericOp, bool isTernary)
static bool	findIndexOfScalarOperand (GenericOp genericOp, int &index)
static std::optional< ElementwiseKind >	getElementwiseKind (Operation *op)
static FailureOr< LinalgOp >	specializeLinalgElementwise (RewriterBase &rewriter, 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);      \

DEBUG_TYPE

#define DEBUG_TYPE   "linalg-specialization"

Definition at line 30 of file Specialize.cpp.

GEN_PASS_DEF_LINALGSPECIALIZEGENERICOPSPASS

#define GEN_PASS_DEF_LINALGSPECIALIZEGENERICOPSPASS

Definition at line 26 of file Specialize.cpp.

Function Documentation

areBinOpsSwapped()

bool areBinOpsSwapped ( GenericOp genericOp, bool isTernary )

static

Definition at line 50 of file Specialize.cpp.

References mlir::Block::front(), mlir::IROperand< DerivedT, IRValueT >::get(), mlir::Block::getArgument(), and mlir::Operation::getOpOperand().

Referenced by specializeLinalgElementwise().

findIndexOfScalarOperand()

bool findIndexOfScalarOperand ( GenericOp genericOp, int & index )

static

Definition at line 84 of file Specialize.cpp.

References mlir::Block::front(), and mlir::Operation::getOperands().

Referenced by specializeLinalgElementwise().

getElementwiseKind()

std::optional< ElementwiseKind > getElementwiseKind ( Operation * op )

static

Definition at line 101 of file Specialize.cpp.

Referenced by specializeLinalgElementwise().

specializeLinalgElementwise()

FailureOr< LinalgOp > specializeLinalgElementwise ( RewriterBase & rewriter, GenericOp genericOp )

static

Definition at line 172 of file Specialize.cpp.

References areBinOpsSwapped(), findIndexOfScalarOperand(), mlir::AffineMap::get(), mlir::Builder::getAffineMapArrayAttr(), mlir::linalg::getArityGroupAndKind(), mlir::Builder::getContext(), getElementwiseKind(), mlir::Operation::getNumOperands(), mlir::Operation::getOperand(), mlir::Operation::getOperands(), mlir::RewriterBase::notifyMatchFailure(), and mlir::RewriterBase::replaceOp().

Referenced by mlir::linalg::specializeGenericOp().