LoopFusionUtils.cpp File Reference

#include "mlir/Dialect/Affine/LoopFusionUtils.h"
#include "mlir/Analysis/SliceAnalysis.h"
#include "mlir/Dialect/Affine/Analysis/AffineAnalysis.h"
#include "mlir/Dialect/Affine/Analysis/LoopAnalysis.h"
#include "mlir/Dialect/Affine/Analysis/Utils.h"
#include "mlir/Dialect/Affine/IR/AffineOps.h"
#include "mlir/Dialect/Affine/LoopUtils.h"
#include "mlir/IR/IRMapping.h"
#include "mlir/IR/Operation.h"
#include "llvm/Support/Debug.h"
#include "llvm/Support/raw_ostream.h"
#include <optional>

Go to the source code of this file.

Macros
#define	DEBUG_TYPE   "loop-fusion-utils"

Functions
static void	getLoadAndStoreMemRefAccesses (Operation *opA, DenseMap< Value, bool > &values)
static bool	isDependentLoadOrStoreOp (Operation *op, DenseMap< Value, bool > &values)
	Returns true if 'op' is a load or store operation which access a memref accessed 'values' and at least one of the access is a store operation. More...
static Operation *	getFirstDependentOpInRange (Operation *opA, Operation *opB)
static Operation *	getLastDependentOpInRange (Operation *opA, Operation *opB)
static Operation *	getFusedLoopNestInsertionPoint (AffineForOp srcForOp, AffineForOp dstForOp)
static bool	gatherLoadsAndStores (AffineForOp forOp, SmallVectorImpl< Operation * > &loadAndStoreOps)
static unsigned	getMaxLoopDepth (ArrayRef< Operation * > srcOps, ArrayRef< Operation * > dstOps)
	Returns the maximum loop depth at which we could fuse producer loop 'srcForOp' into consumer loop 'dstForOp' without violating data dependences. More...
static LogicalResult	promoteSingleIterReductionLoop (AffineForOp forOp, bool siblingFusionUser)
	Patch the loop body of a forOp that is a single iteration reduction loop into its containing block. More...
static int64_t	getComputeCostHelper (Operation *forOp, LoopNestStats &stats, llvm::SmallDenseMap< Operation *, uint64_t, 8 > *tripCountOverrideMap, DenseMap< Operation *, int64_t > *computeCostMap)

Macro Definition Documentation

DEBUG_TYPE

#define DEBUG_TYPE   "loop-fusion-utils"

Definition at line 26 of file LoopFusionUtils.cpp.

Function Documentation

gatherLoadsAndStores()

static bool gatherLoadsAndStores ( AffineForOp  forOp, SmallVectorImpl< Operation * > &  loadAndStoreOps  )

static

Definition at line 170 of file LoopFusionUtils.cpp.

Referenced by mlir::affine::canFuseLoops().

getComputeCostHelper()

static int64_t getComputeCostHelper ( Operation *  forOp, LoopNestStats &  stats, llvm::SmallDenseMap< Operation *, uint64_t, 8 > *  tripCountOverrideMap, DenseMap< Operation *, int64_t > *  computeCostMap  )

static

Definition at line 523 of file LoopFusionUtils.cpp.

References mlir::affine::LoopNestStats::loopMap, mlir::affine::LoopNestStats::opCountMap, and mlir::affine::LoopNestStats::tripCountMap.

Referenced by mlir::affine::getComputeCost().

getFirstDependentOpInRange()

static Operation* getFirstDependentOpInRange ( Operation *  opA, Operation *  opB  )

static

Definition at line 61 of file LoopFusionUtils.cpp.

References getLoadAndStoreMemRefAccesses(), and mlir::Operation::walk().

Referenced by getFusedLoopNestInsertionPoint().

getFusedLoopNestInsertionPoint()

static Operation* getFusedLoopNestInsertionPoint ( AffineForOp  srcForOp, AffineForOp  dstForOp  )

static

Definition at line 132 of file LoopFusionUtils.cpp.

References getFirstDependentOpInRange(), getLastDependentOpInRange(), and mlir::Operation::isBeforeInBlock().

Referenced by mlir::affine::canFuseLoops().

getLastDependentOpInRange()

static Operation* getLastDependentOpInRange ( Operation *  opA, Operation *  opB  )

static

Definition at line 87 of file LoopFusionUtils.cpp.

References getLoadAndStoreMemRefAccesses(), and mlir::Operation::walk().

Referenced by getFusedLoopNestInsertionPoint().

getLoadAndStoreMemRefAccesses()

static void getLoadAndStoreMemRefAccesses ( Operation *  opA, DenseMap< Value, bool > &  values  )

static

Definition at line 33 of file LoopFusionUtils.cpp.

References mlir::Operation::walk().

Referenced by getFirstDependentOpInRange(), and getLastDependentOpInRange().

getMaxLoopDepth()

static unsigned getMaxLoopDepth ( ArrayRef< Operation * >  srcOps, ArrayRef< Operation * >  dstOps  )

static

Returns the maximum loop depth at which we could fuse producer loop 'srcForOp' into consumer loop 'dstForOp' without violating data dependences.

Definition at line 186 of file LoopFusionUtils.cpp.

References mlir::affine::gatherProducerConsumerMemrefs(), and mlir::affine::getInnermostCommonLoopDepth().

Referenced by mlir::affine::canFuseLoops().

isDependentLoadOrStoreOp()

static bool isDependentLoadOrStoreOp ( Operation *  op, DenseMap< Value, bool > &  values  )

static

Returns true if 'op' is a load or store operation which access a memref accessed 'values' and at least one of the access is a store operation.

Returns false otherwise.

Definition at line 48 of file LoopFusionUtils.cpp.

promoteSingleIterReductionLoop()

static LogicalResult promoteSingleIterReductionLoop ( AffineForOp  forOp, bool  siblingFusionUser  )

static

Patch the loop body of a forOp that is a single iteration reduction loop into its containing block.

Definition at line 358 of file LoopFusionUtils.cpp.

References mlir::failure(), mlir::affine::getConstantTripCount(), mlir::getForwardSlice(), mlir::affine::replaceForOpWithNewYields(), and mlir::topologicalSort().

Referenced by mlir::affine::fuseLoops().