Home
last modified time | relevance | path

Searched refs:getUnrollFactor (Results 1 – 2 of 2) sorted by relevance

/llvm-project-15.0.7/mlir/lib/Dialect/Affine/Transforms/
H A DLoopUnroll.cpp40 const std::function<unsigned(AffineForOp)> getUnrollFactor; member
42 LoopUnroll() : getUnrollFactor(nullptr) {} in LoopUnroll()
49 const std::function<unsigned(AffineForOp)> &getUnrollFactor = nullptr) in LoopUnroll()
50 : getUnrollFactor(getUnrollFactor) { in LoopUnroll()
106 for (unsigned i = 0; i < numRepetitions || getUnrollFactor; i++) { in runOnOperation()
124 if (getUnrollFactor) in runOnAffineForOp()
125 return loopUnrollByFactor(forOp, getUnrollFactor(forOp)); in runOnAffineForOp()
137 const std::function<unsigned(AffineForOp)> &getUnrollFactor) { in createLoopUnrollPass() argument
140 unrollUpToFactor, unrollFull, getUnrollFactor); in createLoopUnrollPass()
/llvm-project-15.0.7/mlir/include/mlir/Dialect/Affine/
H A DPasses.h94 const std::function<unsigned(AffineForOp)> &getUnrollFactor = nullptr);