aboutsummaryrefslogtreecommitdiff
path: root/llvm/lib/Target
diff options
context:
space:
mode:
authormuneebkhan85 <150162960+muneebkhan85@users.noreply.github.com>2024-06-21 15:39:43 +0100
committerGitHub <noreply@github.com>2024-06-21 16:39:43 +0200
commita9efcbf490d9b8f46ec37062ca8653b4068000e5 (patch)
treeea0ee87aa6c308e7df34e6a969d1d146d904e457 /llvm/lib/Target
parent74a105ad80725b4a54ef76950c938ffe76796b3b (diff)
downloadllvm-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