Searched refs:unrollFactor (Results 1 – 8 of 8) sorted by relevance
| /llvm-project-15.0.7/mlir/lib/Dialect/Affine/Transforms/ |
| H A D | LoopUnroll.cpp | 47 Optional<unsigned> unrollFactor = None, bool unrollUpToFactor = false, in LoopUnroll()
51 if (unrollFactor) in LoopUnroll()
52 this->unrollFactor = *unrollFactor; in LoopUnroll()
131 return loopUnrollUpToFactor(forOp, unrollFactor); in runOnAffineForOp()
132 return loopUnrollByFactor(forOp, unrollFactor); in runOnAffineForOp()
136 int unrollFactor, bool unrollUpToFactor, bool unrollFull, in createLoopUnrollPass() argument
139 unrollFactor == -1 ? None : Optional<unsigned>(unrollFactor), in createLoopUnrollPass()
|
| /llvm-project-15.0.7/mlir/test/lib/Dialect/SCF/ |
| H A D | TestLoopUnrolling.cpp | 46 unrollFactor = unrollFactorParam; in TestLoopUnrollingPass()
67 (void)loopUnrollByFactor(loop, unrollFactor, annotateFn); in runOnOperation()
69 Option<uint64_t> unrollFactor{*this, "unroll-factor", member
|
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/Affine/ |
| H A D | LoopUtils.h | 49 AffineForOp forOp, uint64_t unrollFactor,
54 LogicalResult loopUnrollUpToFactor(AffineForOp forOp, uint64_t unrollFactor);
|
| H A D | Passes.h | 92 int unrollFactor = -1, bool unrollUpToFactor = false,
|
| H A D | Passes.td | 204 Option<"unrollFactor", "unroll-factor", "unsigned", /*default=*/"4",
|
| /llvm-project-15.0.7/mlir/lib/Dialect/SCF/Utils/ |
| H A D | Utils.cpp | 366 Block *loopBodyBlock, Value forOpIV, uint64_t unrollFactor, in generateUnrolledLoop() argument
384 for (unsigned i = 1; i < unrollFactor; i++) { in generateUnrolledLoop()
419 scf::ForOp forOp, uint64_t unrollFactor, in loopUnrollByFactor() argument
421 assert(unrollFactor > 0 && "expected positive unroll factor"); in loopUnrollByFactor()
448 if (unrollFactor == 1) { in loopUnrollByFactor()
454 int64_t tripCountEvenMultiple = tripCount - (tripCount % unrollFactor); in loopUnrollByFactor()
457 int64_t stepUnrolledCst = stepCst * unrollFactor; in loopUnrollByFactor()
482 boundsBuilder.create<arith::ConstantIndexOp>(loc, unrollFactor); in loopUnrollByFactor()
525 forOp.getBody(), forOp.getInductionVar(), unrollFactor, in loopUnrollByFactor()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Affine/Utils/ |
| H A D | LoopUtils.cpp | 56 getCleanupLoopLowerBound(AffineForOp forOp, unsigned unrollFactor, in getCleanupLoopLowerBound() argument
990 uint64_t unrollFactor) { in loopUnrollUpToFactor() argument
993 mayBeConstantTripCount.value() < unrollFactor) in loopUnrollUpToFactor()
995 return loopUnrollByFactor(forOp, unrollFactor); in loopUnrollUpToFactor()
1003 Block *loopBodyBlock, Value forOpIV, uint64_t unrollFactor, in generateUnrolledLoop() argument
1021 for (unsigned i = 1; i < unrollFactor; i++) { in generateUnrolledLoop()
1092 AffineForOp forOp, uint64_t unrollFactor, in loopUnrollByFactor() argument
1097 if (unrollFactor == 1) { in loopUnrollByFactor()
1114 if (getLargestDivisorOfTripCount(forOp) % unrollFactor != 0) { in loopUnrollByFactor()
1132 forOp.setStep(step * unrollFactor); in loopUnrollByFactor()
[all …]
|
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/SCF/Utils/ |
| H A D | Utils.h | 136 scf::ForOp forOp, uint64_t unrollFactor,
|