Specialize.cpp File Reference

#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 (isaConvolutionOpOfType<ConvOpTy>(genericOp, &dilations, &strides))       \
    return specializeToConvOp<ConvOpTy>(rewriter, genericOp, dilations,        \
                                        strides);                              \

DEBUG_TYPE

#define DEBUG_TYPE   "linalg-specialization"

Definition at line 28 of file Specialize.cpp.

GEN_PASS_DEF_LINALGSPECIALIZEGENERICOPSPASS

#define GEN_PASS_DEF_LINALGSPECIALIZEGENERICOPSPASS

Definition at line 24 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]}))

Definition at line 30 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 37 of file Specialize.cpp.

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

Function Documentation

areBinOpsSwapped()

bool areBinOpsSwapped ( GenericOp genericOp )

static

Definition at line 56 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().