Searched refs:tileSizePerThread (Results 1 – 1 of 1) sorted by relevance
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/ |
| H A D | Tiling.cpp | 282 OpFoldResult tileSizePerThread = in tileToForeachThreadOpImpl() local
291 b, loc, i + j * M, {offset, threadId, tileSizePerThread}); in tileToForeachThreadOpImpl()
295 {offset, nonZeroNumThreads[threadIdIdx], tileSizePerThread, size}); in tileToForeachThreadOpImpl()
299 tileSizePerThread = makeComposedFoldedAffineMin( in tileToForeachThreadOpImpl()
301 ArrayRef<OpFoldResult>{sizeMinusOffsetPerThread, tileSizePerThread}); in tileToForeachThreadOpImpl()
307 !canOmitTileOffsetInBoundsCheck(tileSizePerThread, in tileToForeachThreadOpImpl()
309 tileSizePerThread = buildMax(b, loc, {zero, tileSizePerThread}); in tileToForeachThreadOpImpl()
311 tiledSizes.push_back(tileSizePerThread); in tileToForeachThreadOpImpl()
|