MLIR  16.0.0git
Macros | Functions
LoopUnroll.cpp File Reference
#include "PassDetail.h"
#include "mlir/Dialect/Affine/Analysis/LoopAnalysis.h"
#include "mlir/Dialect/Affine/IR/AffineOps.h"
#include "mlir/Dialect/Affine/LoopUtils.h"
#include "mlir/Dialect/Affine/Passes.h"
#include "mlir/IR/AffineExpr.h"
#include "mlir/IR/AffineMap.h"
#include "mlir/IR/Builders.h"
#include "llvm/ADT/DenseMap.h"
#include "llvm/Support/CommandLine.h"
#include "llvm/Support/Debug.h"
+ Include dependency graph for LoopUnroll.cpp:

Go to the source code of this file.

Macros

#define DEBUG_TYPE   "affine-loop-unroll"
 

Functions

static bool isInnermostAffineForOp (AffineForOp op)
 Returns true if no other affine.for ops are nested within op. More...
 
static void gatherInnermostLoops (func::FuncOp f, SmallVectorImpl< AffineForOp > &loops)
 Gathers loops that have no affine.for's nested within. More...
 

Macro Definition Documentation

◆ DEBUG_TYPE

#define DEBUG_TYPE   "affine-loop-unroll"

Definition at line 26 of file LoopUnroll.cpp.

Function Documentation

◆ gatherInnermostLoops()

static void gatherInnermostLoops ( func::FuncOp  f,
SmallVectorImpl< AffineForOp > &  loops 
)
static

Gathers loops that have no affine.for's nested within.

Definition at line 74 of file LoopUnroll.cpp.

References mlir::getConstantTripCount(), isInnermostAffineForOp(), mlir::loopUnrollByFactor(), mlir::loopUnrollFull(), mlir::loopUnrollUpToFactor(), and mlir::succeeded().

◆ isInnermostAffineForOp()

static bool isInnermostAffineForOp ( AffineForOp  op)
static

Returns true if no other affine.for ops are nested within op.

Definition at line 65 of file LoopUnroll.cpp.

References mlir::WalkResult::interrupt().

Referenced by gatherInnermostLoops().