Searched refs:LoopReduction (Results 1 – 9 of 9) sorted by relevance
| /llvm-project/mlir/lib/Dialect/Affine/Transforms/ |
| H A D | AffineParallelize.cpp | 50 ParallelizationCandidate(AffineForOp l, SmallVector<LoopReduction> &&r) in ParallelizationCandidate()
56 SmallVector<LoopReduction> reductions;
67 SmallVector<LoopReduction> reductions; in runOnOperation()
|
| H A D | SuperVectorize.cpp | 1328 ArrayRef<LoopReduction> reductions; in vectorizeAffineForOp()
1772 SmallVector<LoopReduction, 2> reductions; in runOnOperation()
|
| /llvm-project/mlir/include/mlir/Dialect/Affine/Analysis/ |
| H A D | AffineAnalysis.h | 35 struct LoopReduction { struct
48 AffineForOp forOp, SmallVectorImpl<LoopReduction> &supportedReductions); argument
55 SmallVectorImpl<LoopReduction> *parallelReductions = nullptr);
|
| /llvm-project/mlir/include/mlir/Dialect/Affine/ |
| H A D | Utils.h | 42 using ReductionLoopMap = DenseMap<Operation *, SmallVector<LoopReduction, 2>>;
52 ArrayRef<LoopReduction> parallelReductions = {},
|
| /llvm-project/mlir/lib/Dialect/Affine/Analysis/ |
| H A D | AffineAnalysis.cpp | 86 AffineForOp forOp, SmallVectorImpl<LoopReduction> &supportedReductions) { in getSupportedReductions()
94 supportedReductions.emplace_back(LoopReduction{kind, i, value}); in getSupportedReductions()
102 AffineForOp forOp, SmallVectorImpl<LoopReduction> *parallelReductions) { in isLoopParallel()
|
| H A D | Utils.cpp | 1963 SmallVector<LoopReduction> reductions; in isLoopParallelAndContainsReduction()
|
| /llvm-project/mlir/test/lib/Dialect/Affine/ |
| H A D | TestVectorizationUtils.cpp | 243 SmallVector<LoopReduction, 2> reductions; in testVecAffineLoopNest()
|
| /llvm-project/mlir/lib/Dialect/Affine/Utils/ |
| H A D | Utils.cpp | 353 ArrayRef<LoopReduction> parallelReductions, in affineParallelize()
369 parallelReductions, [](const LoopReduction &red) { return red.value; })); in affineParallelize()
371 parallelReductions, [](const LoopReduction &red) { return red.kind; })); in affineParallelize()
|
| H A D | LoopUtils.cpp | 1139 SmallVector<LoopReduction> reductions; in loopUnrollJamByFactor()
1268 for (LoopReduction &reduction : reductions) { in loopUnrollJamByFactor()
|