Home
last modified time | relevance | path

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

12

/llvm-project-15.0.7/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()
334 SmallVector<Value> blockEndCoord(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()
735 SmallVector<Value> tripCounts(op.getNumLoops()); in matchAndRewrite()
736 for (size_t i = 0; i < op.getNumLoops(); ++i) { in matchAndRewrite()
793 SmallVector<int64_t> numIterations(op.getNumLoops()); in matchAndRewrite()
[all …]
/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/
H A DInterchange.cpp41 genericOp.getNumLoops() != interchangeVector.size()) in interchangeGenericOpPrecondition()
96 allIndices.reserve(genericOp.getNumLoops()); in interchangeGenericOp()
97 llvm::transform(llvm::seq<uint64_t>(0, genericOp.getNumLoops()), in interchangeGenericOp()
H A DFusionOnTensors.cpp154 SmallVector<Value> tileIvs(producerOp.getNumLoops(), nullptr); in getTiledProducer()
155 SmallVector<Value> tileSizes(producerOp.getNumLoops(), zero); in getTiledProducer()
156 SmallVector<Value> allIvs(producerOp.getNumLoops(), nullptr); in getTiledProducer()
H A DElementwiseOpFusion.cpp76 if (producer.getNumParallelLoops() != producer.getNumLoops()) in areElementwiseOpsFusable()
87 if (consumerIndexMap.getNumResults() != producer.getNumLoops()) in areElementwiseOpsFusable()
106 BitVector coveredDims(consumer.getNumLoops(), false); in areElementwiseOpsFusable()
159 std::max(producer.getNumLoops(), consumer.getNumLoops()); in generateFusedElementwiseOpRegion()
349 consumerOpOperand, consumer.getNumLoops()); in fuseElementwiseOpsImpl()
1336 if (genericOp.getNumLoops() <= 1 || foldedIterationDims.empty() || in collapseGenericOpIterationDims()
1343 if (failed(collapsingInfo.initialize(genericOp.getNumLoops(), in collapseGenericOpIterationDims()
H A DFusePadOpWithLinalgProducer.cpp60 if (linalgOp.getNumLoops() != linalgOp.getNumParallelLoops()) { in matchAndRewrite()
H A DTiling.cpp77 SmallVector<Value> allIvs(op.getNumLoops(), nullptr); in transformIndexOps()
111 if (dimension >= op.getNumLoops()) in computeMultiTileSizes()
386 auto nLoops = op.getNumLoops(); in tileLinalgOpImpl()
521 auto nLoops = op.getNumLoops(); in tileLinalgOpImpl()
H A DTransforms.cpp485 if (options.tileSizes.size() < rootOp.getNumLoops()) in returningMatchAndRewrite()
497 rootOp.getNumLoops()); in returningMatchAndRewrite()
500 ? llvm::to_vector<6>(llvm::seq<int64_t>(0, rootOp.getNumLoops())) in returningMatchAndRewrite()
503 rootOp.getNumLoops()); in returningMatchAndRewrite()
H A DSplit.cpp101 if (dimension >= op.getNumLoops()) in splitOp()
H A DTilingInterfaceImpl.cpp156 auto numLoops = linalgOp.getNumLoops(); in generateResultTileValue()
H A DDecomposeLinalgOps.cpp251 if (genericOp.getNumParallelLoops() != genericOp.getNumLoops()) { in matchAndRewrite()
H A DLoops.cpp185 assert(linalgOp.getNumLoops() == allIvs.size() && in replaceIndexOpsByInductionVariables()
H A DFusion.cpp122 for (unsigned i = 0, e = producer.getNumLoops(); i < e; ++i) { in fuse()
H A DSplitReduction.cpp363 auto nDims = op.getNumLoops() + 1; in splitReductionByScaling()
/llvm-project-15.0.7/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-15.0.7/mlir/lib/Dialect/GPU/Transforms/
H A DParallelLoopMapper.cpp114 attrs.reserve(parallelOp.getNumLoops()); in mapParallelOp()
115 for (int i = 0, e = parallelOp.getNumLoops(); i < e; ++i) { in mapParallelOp()
/llvm-project-15.0.7/mlir/lib/Dialect/SCF/Transforms/
H A DParallelLoopFusion.cpp35 if (firstPloop.getNumLoops() != secondPloop.getNumLoops()) in equalIterationSpaces()
H A DLoopCanonicalization.cpp165 for (unsigned idx = 0; idx < parOp.getNumLoops(); ++idx) { in matchAndRewrite()
/llvm-project-15.0.7/mlir/lib/Dialect/SCF/TransformOps/
H A DSCFTransformOps.cpp42 for (unsigned i = 0, e = getNumLoops(); i < e; ++i) { in apply()
/llvm-project-15.0.7/mlir/lib/Dialect/SparseTensor/Transforms/
H A DSparseTensorRewriting.cpp122 op.getNumParallelLoops() != op.getNumLoops() || in matchAndRewrite()
/llvm-project-15.0.7/mlir/lib/Target/LLVMIR/Dialect/OpenMP/
H A DOpenMPToLLVMIRTranslation.cpp788 if (loopInfos.size() != loop.getNumLoops() - 1) in convertOmpWsLoop()
802 for (unsigned i = 0, e = loop.getNumLoops(); i < e; ++i) { in convertOmpWsLoop()
930 if (loopInfos.size() != loop.getNumLoops() - 1) in convertOmpSimdLoop()
944 for (unsigned i = 0, e = loop.getNumLoops(); i < e; ++i) { in convertOmpSimdLoop()
/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/IR/
H A DLinalgInterfaces.cpp398 if (allLoopDims.size() != linalgOp.getNumLoops()) in isConvolutionInterfaceImpl()
664 unsigned numLoops = linalgOp.getNumLoops(); in verifyStructuredOpInterface()
H A DLinalgOps.cpp1528 if (linalgOp.getNumLoops() <= dim()) in verify()
1531 << linalgOp.getNumLoops() << ") of the enclosing LinalgOp"; in verify()
/llvm-project-15.0.7/mlir/lib/Conversion/SCFToControlFlow/
H A DSCFToControlFlow.cpp458 ivs.reserve(parallelOp.getNumLoops()); in matchAndRewrite()
/llvm-project-15.0.7/mlir/test/lib/Dialect/Linalg/
H A DTestLinalgTransforms.cpp471 unsigned insertDimIndex = op.getNumLoops() - 1; in applySplitReduction()
/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/TransformOps/
H A DLinalgTransformOps.cpp735 if (getDimension() >= linalgOp.getNumLoops()) { in apply()

12