Home
last modified time | relevance | path

Searched refs:distributionMethod (Results 1 – 3 of 3) sorted by relevance

/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Utils/
H A DUtils.cpp511 SmallVector<DistributionMethod, 0> distributionMethod; in doit() local
520 distributionMethod = distributionOptions->distributionMethod; in doit()
647 if (distributionMethod.empty()) {
659 (distributionMethod.size() < nOuterPar) in __anon19d834b50e02()
667 DistributionMethod methodToUse = distributionMethod[0];
670 if (distributionMethod[i] != methodToUse)
688 (distributionMethod.size() < numProcessed) in __anon19d834b50f02()
706 distributionMethod.drop_front(numProcessed)); in __anon19d834b51002()
750 SmallVector<DistributionMethod, 0> distributionMethod; in doit() local
753 distributionMethod.assign(distributionOptions->distributionMethod.begin(), in doit()
[all …]
/llvm-project-15.0.7/mlir/test/lib/Dialect/Linalg/
H A DTestLinalgTransforms.cpp268 cyclicNprocsEqNiters.distributionMethod.resize( in fillTileAndDistributePatterns()
285 cyclicNprocsGeNiters.distributionMethod.resize( in fillTileAndDistributePatterns()
302 cyclicNprocsDefault.distributionMethod.resize(2, in fillTileAndDistributePatterns()
319 cyclicNprocsMixed1.distributionMethod = { in fillTileAndDistributePatterns()
336 cyclicNprocsMixed2.distributionMethod = { in fillTileAndDistributePatterns()
353 cyclicNprocsMixed3.distributionMethod = { in fillTileAndDistributePatterns()
371 cyclicNprocsEqNiters.distributionMethod.resize(2, in fillTileAndDistributePatterns()
390 cyclicNprocsEqNiters.distributionMethod.resize(2, DistributionMethod::Cyclic); in fillTileFuseAndDistributePatterns()
/llvm-project-15.0.7/mlir/include/mlir/Dialect/Linalg/Utils/
H A DUtils.h363 SmallVector<DistributionMethod, 0> distributionMethod = {}; member