Searched refs:parallelReductions (Results 1 – 6 of 6) sorted by relevance
| /llvm-project/mlir/lib/Dialect/Affine/Analysis/ |
| H A D | AffineAnalysis.cpp | 102 AffineForOp forOp, SmallVectorImpl<LoopReduction> *parallelReductions) { in isLoopParallel() argument
107 if (numIterArgs > 0 && !parallelReductions) in isLoopParallel()
111 if (parallelReductions) { in isLoopParallel()
112 getSupportedReductions(forOp, *parallelReductions); in isLoopParallel()
115 if (parallelReductions->size() != numIterArgs) in isLoopParallel()
|
| /llvm-project/mlir/lib/Dialect/Affine/Transforms/ |
| H A D | AffineParallelize.cpp | 68 if (isLoopParallel(loop, parallelReductions ? &reductions : nullptr)) in runOnOperation()
|
| /llvm-project/mlir/include/mlir/Dialect/Affine/Analysis/ |
| H A D | AffineAnalysis.h | 55 SmallVectorImpl<LoopReduction> *parallelReductions = nullptr);
|
| /llvm-project/mlir/include/mlir/Dialect/Affine/ |
| H A D | Utils.h | 47 /// reductions specified in `parallelReductions` are also parallelized.
49 /// are not listed in `parallelReductions`. `resOp` if non-null is set to the
52 ArrayRef<LoopReduction> parallelReductions = {},
|
| H A D | Passes.td | 378 Option<"parallelReductions", "parallel-reductions", "bool",
|
| /llvm-project/mlir/lib/Dialect/Affine/Utils/ |
| H A D | Utils.cpp | 353 ArrayRef<LoopReduction> parallelReductions, in affineParallelize()
356 unsigned numReductions = parallelReductions.size(); in affineParallelize()
369 parallelReductions, [](const LoopReduction &red) { return red.value; })); in affineParallelize()
371 parallelReductions, [](const LoopReduction &red) { return red.kind; })); in affineParallelize()
349 affineParallelize(AffineForOp forOp,ArrayRef<LoopReduction> parallelReductions,AffineParallelOp * resOp) affineParallelize() argument
|