Commit d386951
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- test/Conversion/intel
- third_party/intel/lib/TritonIntelGPUToLLVM
2 files changed
+613
-5
lines changed
0 commit comments