Home
last modified time | relevance | path

Searched refs:getNumLoops (Results 1 – 25 of 50) sorted by relevance

12

/llvm-project/mlir/lib/Dialect/Async/Transforms/
H A DAsyncParallelFor.cpp218 inputs.reserve(2 + 4 * op.getNumLoops() + captures.size()); in getParallelComputeFunctionType()
227 for (unsigned i = 0; i < op.getNumLoops(); ++i) in getParallelComputeFunctionType()
233 for (unsigned i = 0; i < op.getNumLoops(); ++i) { in getParallelComputeFunctionType()
279 ParallelComputeFunctionArgs args = {op.getNumLoops(), func.getArguments()}; in createParallelComputeFunction()
334 SmallVector<Value> blockEndCoord(op.getNumLoops()); in createParallelComputeFunction()
349 SmallVector<Value> computeBlockInductionVars(op.getNumLoops()); in createParallelComputeFunction()
368 SmallVector<Value> isBlockFirstCoord(op.getNumLoops()); in createParallelComputeFunction()
369 SmallVector<Value> isBlockLastCoord(op.getNumLoops()); in createParallelComputeFunction()
397 if (loopIdx < op.getNumLoops() - 1) { in createParallelComputeFunction()
398 if (loopIdx + 1 >= op.getNumLoops() in createParallelComputeFunction()
[all...]
/llvm-project/mlir/lib/Dialect/Linalg/Transforms/
H A DInterchange.cpp40 genericOp.getNumLoops() != interchangeVector.size()) in interchangeGenericOpPrecondition()
93 allIndices.reserve(genericOp.getNumLoops()); in interchangeGenericOp()
94 llvm::transform(llvm::seq<uint64_t>(0, genericOp.getNumLoops()), in interchangeGenericOp()
H A DEliminateEmptyTensors.cpp47 if (op.getNumParallelLoops() != op.getNumLoops()) in linalgOpAnchoredEmptyTensorEliminationStep()
H A DTiling.cpp81 SmallVector<Value> allIvs(op.getNumLoops(), nullptr); in transformIndexOps()
118 assert(dimension < op.getNumLoops() && "dimension overflow"); in computeStaticContinuousTileSizes()
248 assert(dimension < op.getNumLoops() && "dimension overflow"); in computeStaticMultiTileSizes()
272 if (dimension >= op.getNumLoops()) in computeMultiTileSizes()
442 auto nLoops = op.getNumLoops();
809 auto nLoops = op.getNumLoops(); in tileReductionUsingForall()
H A DFusePadOpWithLinalgProducer.cpp60 if (linalgOp.getNumLoops() != linalgOp.getNumParallelLoops()) { in matchAndRewrite()
H A DElementwiseOpFusion.cpp95 return producer.getNumLoops() == 0 && consumer.getNumLoops() == 0; in areElementwiseOpsFusable()
158 if (producer.getNumParallelLoops() != producer.getNumLoops()) in areElementwiseOpsFusable()
169 if (consumerIndexMap.getNumResults() != producer.getNumLoops())
184 BitVector coveredDims(consumer.getNumLoops(), false); in generateFusedElementwiseOpRegion()
234 std::max(producer.getNumLoops(), consumer.getNumLoops()); in generateFusedElementwiseOpRegion()
449 consumer.getNumLoops(), preservedProducerResults); in matchAndRewrite()
1679 if (op.getNumLoops() <= 1 || foldedIterationDims.empty() || in collapseOpIterationDims()
1700 collapsingInfo.initialize(op.getNumLoops(), foldedIterationDim in collapseOpIterationDims()
[all...]
H A DBufferizableOpInterfaceImpl.cpp125 if (linalgOp.getNumLoops() != linalgOp.getNumParallelLoops()) in bufferizesToElementwiseAccess()
H A DFusion.cpp119 for (unsigned i = 0, e = producer.getNumLoops(); i < e; ++i) { in fuse()
H A DDecomposeLinalgOps.cpp253 if (genericOp.getNumParallelLoops() != genericOp.getNumLoops()) { in matchAndRewrite()
H A DLoops.cpp197 assert(linalgOp.getNumLoops() == allIvs.size() && in replaceIndexOpsByInductionVariables()
/llvm-project/llvm/include/llvm/Analysis/
H A DLoopNestAnalysis.h107 for (unsigned I = 0; I < getNumLoops(); ++I) in getLoopIndex()
114 size_t getNumLoops() const { return Loops.size(); } in getNumLoops() function
124 for (unsigned I = 0; I < getNumLoops(); ++I) { in getLoopsAtDepth()
/llvm-project/mlir/lib/Dialect/SparseTensor/Transforms/Utils/
H A DIterationGraphSorter.cpp89 const unsigned numLoops = getNumLoops(); in topoSort()
163 itGraph.resize(getNumLoops(), std::vector<bool>(getNumLoops(), false)); in IterationGraphSorter()
164 inDegree.resize(getNumLoops()); in IterationGraphSorter()
H A DIterationGraphSorter.h54 unsigned getNumLoops() const { return loop2OutLvl.getNumDims(); } in getNumLoops() function
H A DCodegenEnv.h89 unsigned getLoopNum() const { return latticeMerger.getNumLoops(); }
/llvm-project/mlir/lib/Dialect/SCF/Transforms/
H A DParallelLoopCollapsing.cpp95 if (flattenedCombinedLoops.size() != op.getNumLoops()) { in runOnOperation()
97 << op.getNumLoops() in runOnOperation()
H A DParallelLoopFusion.cpp43 if (firstPloop.getNumLoops() != secondPloop.getNumLoops()) in equalIterationSpaces()
/llvm-project/mlir/lib/Dialect/GPU/Transforms/
H A DParallelLoopMapper.cpp120 attrs.reserve(parallelOp.getNumLoops()); in mapParallelOp()
121 for (int i = 0, e = parallelOp.getNumLoops(); i < e; ++i) { in mapParallelOp()
/llvm-project/mlir/lib/Dialect/Linalg/IR/
H A DLinalgInterfaces.cpp55 return linalgOp.getNumLoops() == 0;
187 if (!op.isAllParallelLoops() || op.getNumLoops() < 1) in isContractionBody()
966 if (allLoopDims.size() != linalgOp.getNumLoops()) in verifyFillInterface()
1252 unsigned numLoops = linalgOp.getNumLoops(); in verifyStructuredOpInterface()
/llvm-project/mlir/lib/Dialect/SCF/Utils/
H A DAffineCanonicalizationUtils.cpp42 for (unsigned idx = 0; idx < parOp.getNumLoops(); ++idx) { in matchForLikeLoop()
/llvm-project/mlir/lib/Dialect/Tensor/TransformOps/
H A DTensorTransformOps.cpp195 for (uint64_t i = 0, e = getNumLoops(); i < e; ++i) { in applyToOne()
/llvm-project/mlir/lib/Dialect/Linalg/TransformOps/
H A DLinalgMatchOps.cpp388 getRawDimList(), op.getNumLoops(), dims); in getDimensionsFor()
633 int64_t numLoops = linalgOp.getNumLoops(); in matchOperation()
H A DLinalgTransformOps.cpp1141 unsigned numLoops = cast<LinalgOp>(target.getOperation()).getNumLoops();
1488 if (getMixedPackedSizes().size() != linalgOp.getNumLoops()) {
1491 << getMixedPackedSizes().size() << " vs " << linalgOp.getNumLoops() in verify()
2038 hoistPaddingOnTensors(rewriter, target, getNumLoops(), getTranspose(), in applyToOne()
2360 if (getDimension() >= linalgOp.getNumLoops()) { in apply()
3666 if (target.getNumLoops() <= 1) { in applyToOne()
3672 ReassociationIndices reassociation(target.getNumLoops()); in applyToOne()
/llvm-project/mlir/lib/Dialect/MemRef/TransformOps/
H A DMemRefTransformOps.cpp271 for (uint64_t i = 0, e = getNumLoops(); i < e; ++i) { in applyToOne()
/llvm-project/mlir/lib/Dialect/Tensor/Transforms/
H A DPackAndUnpackPatterns.cpp52 if (linalgOp.getNumParallelLoops() != linalgOp.getNumLoops())
/llvm-project/mlir/include/mlir/Dialect/SparseTensor/Utils/
H A DMerger.h355 constexpr unsigned getNumLoops() const { return numLoops; } in getNumLoops() function

12