Home
last modified time | relevance | path

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

/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Utils/
H A DUtils.cpp506 Optional<LinalgLoopDistributionOptions> distributionOptions, in doit() argument
512 if (distributionOptions) { in doit()
520 distributionMethod = distributionOptions->distributionMethod; in doit()
523 procInfo = distributionOptions->procInfo(b, loc, parallelLoopRanges); in doit()
542 if (!distributionOptions || loopNest.loops.empty()) in doit()
731 Optional<LinalgLoopDistributionOptions> distributionOptions, in doit() argument
751 if (distributionOptions) { in doit()
752 auto &options = *distributionOptions; in doit()
753 distributionMethod.assign(distributionOptions->distributionMethod.begin(), in doit()
754 distributionOptions->distributionMethod.end()); in doit()
/llvm-project-15.0.7/mlir/include/mlir/Dialect/Linalg/Transforms/
H A DTransforms.h584 setDistributionOptions(LinalgLoopDistributionOptions distributionOptions) { in setDistributionOptions()
585 tileDistribution = std::move(distributionOptions); in setDistributionOptions()
638 setDistributionOptions(LinalgLoopDistributionOptions distributionOptions) { in setDistributionOptions()
639 distribution = std::move(distributionOptions); in setDistributionOptions()
/llvm-project-15.0.7/mlir/test/lib/Dialect/Linalg/
H A DTestLinalgTransforms.cpp223 LinalgLoopDistributionOptions distributionOptions; in applyPatterns() local