Home
last modified time | relevance | path

Searched refs:unrollJamFactor (Results 1 – 5 of 5) sorted by relevance

/llvm-project-15.0.7/mlir/lib/Dialect/Affine/Transforms/
H A DLoopUnrollAndJam.cpp57 explicit LoopUnrollAndJam(Optional<unsigned> unrollJamFactor = None) { in LoopUnrollAndJam()
58 if (unrollJamFactor) in LoopUnrollAndJam()
59 this->unrollJamFactor = *unrollJamFactor; in LoopUnrollAndJam()
67 mlir::createLoopUnrollAndJamPass(int unrollJamFactor) { in createLoopUnrollAndJamPass() argument
69 unrollJamFactor == -1 ? None : Optional<unsigned>(unrollJamFactor)); in createLoopUnrollAndJamPass()
81 (void)loopUnrollJamByFactor(forOp, unrollJamFactor); in runOnOperation()
/llvm-project-15.0.7/mlir/include/mlir/Dialect/Affine/
H A DLoopUtils.h72 uint64_t unrollJamFactor);
77 uint64_t unrollJamFactor);
H A DPasses.h100 createLoopUnrollAndJamPass(int unrollJamFactor = -1);
H A DPasses.td222 Option<"unrollJamFactor", "unroll-jam-factor", "unsigned",
/llvm-project-15.0.7/mlir/lib/Dialect/Affine/Utils/
H A DLoopUtils.cpp1153 mayBeConstantTripCount.value() < unrollJamFactor) in loopUnrollJamUpToFactor()
1155 return loopUnrollJamByFactor(forOp, unrollJamFactor); in loopUnrollJamUpToFactor()
1205 if (unrollJamFactor == 1) { in loopUnrollJamByFactor()
1246 if (getLargestDivisorOfTripCount(forOp) % unrollJamFactor != 0) { in loopUnrollJamByFactor()
1276 for (unsigned i = unrollJamFactor - 1; i >= 1; --i) { in loopUnrollJamByFactor()
1295 unsigned oldNumResults = newResults.size() / unrollJamFactor; in loopUnrollJamByFactor()
1298 for (unsigned i = unrollJamFactor - 1; i >= 1; --i) { in loopUnrollJamByFactor()
1313 forOp.setStep(step * unrollJamFactor); in loopUnrollJamByFactor()
1317 for (unsigned i = unrollJamFactor - 1; i >= 1; --i) { in loopUnrollJamByFactor()
1340 newForOp.getNumIterOperands() / unrollJamFactor; in loopUnrollJamByFactor()
[all …]