Home
last modified time | relevance | path

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

/llvm-project-15.0.7/mlir/lib/Dialect/SCF/Utils/
H A DUtils.cpp443 int64_t stepCst = stepCstOp.value(); in loopUnrollByFactor() local
444 assert(lbCst >= 0 && ubCst >= 0 && stepCst >= 0 && in loopUnrollByFactor()
446 int64_t tripCount = mlir::ceilDiv(ubCst - lbCst, stepCst); in loopUnrollByFactor()
455 int64_t upperBoundUnrolledCst = lbCst + tripCountEvenMultiple * stepCst; in loopUnrollByFactor()
457 int64_t stepUnrolledCst = stepCst * unrollFactor; in loopUnrollByFactor()
468 stepUnrolled = stepCst == stepUnrolledCst in loopUnrollByFactor()
552 if (auto stepCst = step.getDefiningOp<arith::ConstantIndexOp>()) in normalizeLoop() local
553 isStepOne = stepCst.value() == 1; in normalizeLoop()
/llvm-project-15.0.7/mlir/lib/Dialect/SCF/Transforms/
H A DLoopPipelining.cpp91 auto stepCst = forOp.getStep().getDefiningOp<arith::ConstantIndexOp>(); in initializeLoopInfo() local
92 if (!upperBoundCst || !lowerBoundCst || !stepCst) in initializeLoopInfo()
96 step = stepCst.value(); in initializeLoopInfo()