/llvm-project/mlir/lib/Dialect/Async/Transforms/ |
H A D | AsyncParallelFor.cpp | 218 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 D | Interchange.cpp | 40 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 D | EliminateEmptyTensors.cpp | 47 if (op.getNumParallelLoops() != op.getNumLoops()) in linalgOpAnchoredEmptyTensorEliminationStep()
|
H A D | Tiling.cpp | 81 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 D | FusePadOpWithLinalgProducer.cpp | 60 if (linalgOp.getNumLoops() != linalgOp.getNumParallelLoops()) { in matchAndRewrite()
|
H A D | ElementwiseOpFusion.cpp | 95 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 D | BufferizableOpInterfaceImpl.cpp | 125 if (linalgOp.getNumLoops() != linalgOp.getNumParallelLoops()) in bufferizesToElementwiseAccess()
|
H A D | Fusion.cpp | 119 for (unsigned i = 0, e = producer.getNumLoops(); i < e; ++i) { in fuse()
|
H A D | DecomposeLinalgOps.cpp | 253 if (genericOp.getNumParallelLoops() != genericOp.getNumLoops()) { in matchAndRewrite()
|
H A D | Loops.cpp | 197 assert(linalgOp.getNumLoops() == allIvs.size() && in replaceIndexOpsByInductionVariables()
|
/llvm-project/llvm/include/llvm/Analysis/ |
H A D | LoopNestAnalysis.h | 107 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 D | IterationGraphSorter.cpp | 89 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 D | IterationGraphSorter.h | 54 unsigned getNumLoops() const { return loop2OutLvl.getNumDims(); } in getNumLoops() function
|
H A D | CodegenEnv.h | 89 unsigned getLoopNum() const { return latticeMerger.getNumLoops(); }
|
/llvm-project/mlir/lib/Dialect/SCF/Transforms/ |
H A D | ParallelLoopCollapsing.cpp | 95 if (flattenedCombinedLoops.size() != op.getNumLoops()) { in runOnOperation() 97 << op.getNumLoops() in runOnOperation()
|
H A D | ParallelLoopFusion.cpp | 43 if (firstPloop.getNumLoops() != secondPloop.getNumLoops()) in equalIterationSpaces()
|
/llvm-project/mlir/lib/Dialect/GPU/Transforms/ |
H A D | ParallelLoopMapper.cpp | 120 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 D | LinalgInterfaces.cpp | 55 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 D | AffineCanonicalizationUtils.cpp | 42 for (unsigned idx = 0; idx < parOp.getNumLoops(); ++idx) { in matchForLikeLoop()
|
/llvm-project/mlir/lib/Dialect/Tensor/TransformOps/ |
H A D | TensorTransformOps.cpp | 195 for (uint64_t i = 0, e = getNumLoops(); i < e; ++i) { in applyToOne()
|
/llvm-project/mlir/lib/Dialect/Linalg/TransformOps/ |
H A D | LinalgMatchOps.cpp | 388 getRawDimList(), op.getNumLoops(), dims); in getDimensionsFor() 633 int64_t numLoops = linalgOp.getNumLoops(); in matchOperation()
|
H A D | LinalgTransformOps.cpp | 1141 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 D | MemRefTransformOps.cpp | 271 for (uint64_t i = 0, e = getNumLoops(); i < e; ++i) { in applyToOne()
|
/llvm-project/mlir/lib/Dialect/Tensor/Transforms/ |
H A D | PackAndUnpackPatterns.cpp | 52 if (linalgOp.getNumParallelLoops() != linalgOp.getNumLoops())
|
/llvm-project/mlir/include/mlir/Dialect/SparseTensor/Utils/ |
H A D | Merger.h | 355 constexpr unsigned getNumLoops() const { return numLoops; } in getNumLoops() function
|