diff options
author | muneebkhan85 <150162960+muneebkhan85@users.noreply.github.com> | 2024-06-21 15:39:43 +0100 |
---|---|---|
committer | GitHub <noreply@github.com> | 2024-06-21 16:39:43 +0200 |
commit | a9efcbf490d9b8f46ec37062ca8653b4068000e5 (patch) | |
tree | ea0ee87aa6c308e7df34e6a969d1d146d904e457 /llvm/lib/Target | |
parent | 74a105ad80725b4a54ef76950c938ffe76796b3b (diff) | |
download | llvm-a9efcbf490d9b8f46ec37062ca8653b4068000e5.zip llvm-a9efcbf490d9b8f46ec37062ca8653b4068000e5.tar.gz llvm-a9efcbf490d9b8f46ec37062ca8653b4068000e5.tar.bz2 |
[MLIR] Add continuous tiling to transform dialect (#82792)
This patch enables continuous tiling of a target structured op using
diminishing tile sizes. In cases where the tensor dimensions are not
exactly divisible by the tile size, we are left with leftover tensor
chunks that are irregularly tiled. This approach enables tiling of the
leftover chunk with a smaller tile size and repeats this process
recursively using exponentially diminishing tile sizes. This eventually
generates a chain of loops that apply tiling using diminishing tile
sizes.
Adds `continuous_tile_sizes` op to the transform dialect. This op, when
given a tile size and a dimension, computes a series of diminishing tile
sizes that can be used to tile the target along the given dimension.
Additionally, this op also generates a series of chunk sizes that the
corresponding tile sizes should be applied to along the given dimension.
Adds `multiway` attribute to `transform.structured.split` that enables
multiway splitting of a single target op along the given dimension, as
specified in a list enumerating the chunk sizes.
Diffstat (limited to 'llvm/lib/Target')
0 files changed, 0 insertions, 0 deletions