Home
last modified time | relevance | path

Searched refs:parallelReductions (Results 1 – 6 of 6) sorted by relevance

/llvm-project-15.0.7/mlir/lib/Dialect/Affine/Analysis/
H A DAffineAnalysis.cpp97 SmallVectorImpl<LoopReduction> *parallelReductions) { in isLoopParallel() argument
102 if (numIterArgs > 0 && !parallelReductions) in isLoopParallel()
106 if (parallelReductions) { in isLoopParallel()
107 getSupportedReductions(forOp, *parallelReductions); in isLoopParallel()
110 if (parallelReductions->size() != numIterArgs) in isLoopParallel()
/llvm-project-15.0.7/mlir/lib/Dialect/Affine/Transforms/
H A DAffineParallelize.cpp58 if (isLoopParallel(loop, parallelReductions ? &reductions : nullptr)) in runOnOperation()
/llvm-project-15.0.7/mlir/include/mlir/Dialect/Affine/Analysis/
H A DAffineAnalysis.h53 SmallVectorImpl<LoopReduction> *parallelReductions = nullptr);
/llvm-project-15.0.7/mlir/include/mlir/Dialect/Affine/
H A DUtils.h47 ArrayRef<LoopReduction> parallelReductions = {});
H A DPasses.td374 Option<"parallelReductions", "parallel-reductions", "bool",
/llvm-project-15.0.7/mlir/lib/Dialect/Affine/Utils/
H A DUtils.cpp349 ArrayRef<LoopReduction> parallelReductions) { in affineParallelize() argument
351 unsigned numReductions = parallelReductions.size(); in affineParallelize()
364 parallelReductions, [](const LoopReduction &red) { return red.value; })); in affineParallelize()
366 parallelReductions, [](const LoopReduction &red) { return red.kind; })); in affineParallelize()