#include "mlir/Dialect/Linalg/Passes.h"
#include "mlir/Dialect/Linalg/IR/Linalg.h"
#include "mlir/Dialect/Linalg/Transforms/Transforms.h"
#include "mlir/Dialect/Linalg/Utils/Utils.h"
#include "mlir/IR/PatternMatch.h"
#include "mlir/Transforms/GreedyPatternRewriteDriver.h"
#include "llvm/ADT/SmallVector.h"
#include <optional>
#include "mlir/Dialect/Linalg/Passes.h.inc"

Classes
class	mlir::impl::LinalgBlockPackMatmulBase< DerivedT >

Namespaces
namespace	mlir
	Include the generated interface declarations.
namespace	mlir::impl
	Attribute collections provide a dictionary-like interface.

Macros
#define	GEN_PASS_DEF_LINALGBLOCKPACKMATMUL

Functions
std::unique_ptr<::mlir::Pass >	mlir::impl::createLinalgBlockPackMatmul ()
std::unique_ptr<::mlir::Pass >	mlir::impl::createLinalgBlockPackMatmul (LinalgBlockPackMatmulOptions options)
std::unique_ptr<::mlir::Pass >	mlir::createLinalgBlockPackMatmul ()
std::unique_ptr<::mlir::Pass >	mlir::createLinalgBlockPackMatmul (LinalgBlockPackMatmulOptions options)
static std::optional< int64_t >	getConstantRange (const Range &range)
	Return constant range span or nullopt, otherwise.
static bool	validateFullTilesOnDims (linalg::LinalgOp linalgOp, ArrayRef< OpFoldResult > tiles, ArrayRef< int64_t > dims)
	Return true if all dimensions are fully divisible by the respective tiles.
static FailureOr< PackTransposeResult >	transposePackedMatmul (RewriterBase &rewriter, linalg::LinalgOp linalgOp, linalg::PackOp packOp, AffineMap operandMap, ArrayRef< unsigned > blocksStartDimPos, bool transposeOuterBlocks, bool transposeInnerBlocks)
	Return failure or packed matmul with one of its operands transposed.

Macro Definition Documentation

◆ GEN_PASS_DEF_LINALGBLOCKPACKMATMUL

#define GEN_PASS_DEF_LINALGBLOCKPACKMATMUL

Definition at line 21 of file BlockPackMatmul.cpp.

Function Documentation

◆ getConstantRange()

std::optional< int64_t > getConstantRange ( const Range & range )

static

Return constant range span or nullopt, otherwise.

Definition at line 29 of file BlockPackMatmul.cpp.

References mlir::getConstantIntValue(), mlir::Range::offset, mlir::Range::size, and mlir::Range::stride.

Referenced by validateFullTilesOnDims().

◆ transposePackedMatmul()

FailureOr< PackTransposeResult > transposePackedMatmul	(	RewriterBase &	rewriter,
		linalg::LinalgOp	linalgOp,
		linalg::PackOp	packOp,
		AffineMap	operandMap,
		ArrayRef< unsigned >	blocksStartDimPos,
		bool	transposeOuterBlocks,
		bool	transposeInnerBlocks )