Skip to content

Commit d386951

Browse files
committed
[TritonIntelGPUToLLVM] Detect sub-group transpose cases when using linear layout
Detect sub-group transpose cases as those in which warp and lane dimensions get swapped and no transfer within block-groups is needed. Use sub-group write operations to store the contents in local memory and vector operations to write back. These will be translated to non-transposed and transposed store and loads respectively. As data is moved within sub-groups, no barriers are needed. For now, handle only the case of a `single sub_group_size^2` block being transposed. This may be split in the future by performing `N*M` iterations for matrices of size `N*sub_group_sizexM*sub_group_size`. Signed-off-by: victor-eds <[email protected]>
1 parent 4355afd commit d386951

File tree

2 files changed

+613
-5
lines changed

2 files changed

+613
-5
lines changed

0 commit comments

Comments
 (0)