Home
last modified time | relevance | path

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

/external/llvm-project/mlir/lib/Dialect/Linalg/Utils/
DUtils.cpp148 distributionOptions->distributionMethod)) in doit()
207 ArrayRef<DistributionMethod> distributionMethod = {}) { in generateParallelLoopNest() argument
230 bodyBuilderFn, ivStorage, distributionMethod); in __anon7ecdf6bc0202()
234 if (distributionMethod.empty()) {
247 (distributionMethod.size() < nOuterPar) in __anon7ecdf6bc0302()
249 : distributionMethod.drop_front(nOuterPar)); in __anon7ecdf6bc0302()
255 DistributionMethod methodToUse = distributionMethod[0];
257 for (unsigned i = 1; i < nOuterPar && i < distributionMethod.size(); ++i) {
258 if (distributionMethod[i] != methodToUse)
277 (distributionMethod.size() < numProcessed) in __anon7ecdf6bc0402()
[all …]
/external/llvm-project/mlir/test/lib/Transforms/
DTestLinalgTransforms.cpp321 cyclicNprocsEqNiters.distributionMethod.resize( in fillTileAndDistributePatterns()
337 cyclicNprocsGeNiters.distributionMethod.resize( in fillTileAndDistributePatterns()
353 cyclicNprocsDefault.distributionMethod.resize(2, in fillTileAndDistributePatterns()
369 cyclicNprocsMixed1.distributionMethod = { in fillTileAndDistributePatterns()
385 cyclicNprocsMixed2.distributionMethod = { in fillTileAndDistributePatterns()
401 cyclicNprocsMixed3.distributionMethod = { in fillTileAndDistributePatterns()
418 cyclicNprocsEqNiters.distributionMethod.resize(2, in fillTileAndDistributePatterns()
/external/llvm-project/mlir/include/mlir/Dialect/Linalg/Utils/
DUtils.h205 SmallVector<DistributionMethod, 0> distributionMethod = {}; member