#include "mlir/Dialect/Linalg/Passes.h.inc"">

MLIR: lib/Dialect/Linalg/Transforms/BlockPackMatmul.cpp File Reference (original) (raw)

#include "[mlir/Dialect/Linalg/Passes.h](Dialect%5F2Linalg%5F2Passes%5F8h%5Fsource.html)"
#include "[mlir/Dialect/Linalg/IR/Linalg.h](mlir%5F2Dialect%5F2Linalg%5F2IR%5F2Linalg%5F8h%5Fsource.html)"
#include "[mlir/Dialect/Linalg/Transforms/Transforms.h](mlir%5F2Dialect%5F2Linalg%5F2Transforms%5F2Transforms%5F8h%5Fsource.html)"
#include "[mlir/Dialect/Linalg/Utils/Utils.h](Dialect%5F2Linalg%5F2Utils%5F2Utils%5F8h%5Fsource.html)"
#include "[mlir/IR/PatternMatch.h](PatternMatch%5F8h%5Fsource.html)"
#include "[mlir/Transforms/GreedyPatternRewriteDriver.h](GreedyPatternRewriteDriver%5F8h%5Fsource.html)"
#include "llvm/ADT/SmallVector.h"
#include "llvm/ADT/TypeSwitch.h"
#include <optional>
#include "mlir/Dialect/Linalg/Passes.h.inc"

Go to the source code of this file.

Namespaces
Functions
static std::optional< int64_t > getConstantRange (const Range &range)
Return constant range span or nullopt, otherwise. More...
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. More...
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. More...

◆ GEN_PASS_DEF_LINALGBLOCKPACKMATMUL

#define GEN_PASS_DEF_LINALGBLOCKPACKMATMUL

◆ getConstantRange()

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

◆ transposePackedMatmul()

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

◆ validateFullTilesOnDims()