Home
last modified time | relevance | path

Searched refs:nLoops (Results 1 – 2 of 2) sorted by relevance

/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/
H A DTiling.cpp259 int64_t nLoops = loopRanges.size(); in tileToForeachThreadOpImpl() local
261 tiledOffsets.reserve(nLoops); in tileToForeachThreadOpImpl()
262 tiledSizes.reserve(nLoops); in tileToForeachThreadOpImpl()
263 for (unsigned loopIdx = 0, threadIdIdx = 0; loopIdx < nLoops; ++loopIdx) { in tileToForeachThreadOpImpl()
351 unsigned nLoops = loopRanges.size(); in tileToForeachThreadOpUsingTileSizes() local
353 numThreads.reserve(nLoops); in tileToForeachThreadOpUsingTileSizes()
386 auto nLoops = op.getNumLoops(); in tileLinalgOpImpl() local
388 tileSizes = tileSizes.take_front(nLoops); in tileLinalgOpImpl()
521 auto nLoops = op.getNumLoops(); in tileLinalgOpImpl() local
524 if (tileSizeVector.size() < nLoops) { in tileLinalgOpImpl()
[all …]
/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Utils/
H A DUtils.cpp628 unsigned nLoops = iteratorTypes.size(); local
630 nLoops - iteratorTypes.drop_while(isParallelIterator).size();