Home
last modified time | relevance | path

Searched refs:LoopReduction (Results 1 – 8 of 8) sorted by relevance

/llvm-project-15.0.7/mlir/lib/Dialect/Affine/Transforms/
H A DAffineParallelize.cpp40 ParallelizationCandidate(AffineForOp l, SmallVector<LoopReduction> &&r) in ParallelizationCandidate()
46 SmallVector<LoopReduction> reductions;
57 SmallVector<LoopReduction> reductions; in runOnOperation()
H A DSuperVectorize.cpp1298 ArrayRef<LoopReduction> reductions; in vectorizeAffineForOp()
1743 SmallVector<LoopReduction, 2> reductions; in runOnOperation()
/llvm-project-15.0.7/mlir/include/mlir/Dialect/Affine/Analysis/
H A DAffineAnalysis.h33 struct LoopReduction { struct
46 AffineForOp forOp, SmallVectorImpl<LoopReduction> &supportedReductions); argument
53 SmallVectorImpl<LoopReduction> *parallelReductions = nullptr);
/llvm-project-15.0.7/mlir/include/mlir/Dialect/Affine/
H A DUtils.h37 using ReductionLoopMap = DenseMap<Operation *, SmallVector<LoopReduction, 2>>;
47 ArrayRef<LoopReduction> parallelReductions = {});
/llvm-project-15.0.7/mlir/lib/Dialect/Affine/Analysis/
H A DAffineAnalysis.cpp81 AffineForOp forOp, SmallVectorImpl<LoopReduction> &supportedReductions) { in getSupportedReductions()
89 supportedReductions.emplace_back(LoopReduction{kind, i, value}); in getSupportedReductions()
97 SmallVectorImpl<LoopReduction> *parallelReductions) { in isLoopParallel()
H A DUtils.cpp1340 SmallVector<LoopReduction> reductions; in isLoopParallelAndContainsReduction()
/llvm-project-15.0.7/mlir/lib/Dialect/Affine/Utils/
H A DUtils.cpp349 ArrayRef<LoopReduction> parallelReductions) { in affineParallelize()
364 parallelReductions, [](const LoopReduction &red) { return red.value; })); in affineParallelize()
366 parallelReductions, [](const LoopReduction &red) { return red.kind; })); in affineParallelize()
H A DLoopUtils.cpp1240 SmallVector<LoopReduction> reductions; in loopUnrollJamByFactor()
1367 for (LoopReduction &reduction : reductions) { in loopUnrollJamByFactor()