llvm-project
9052a85d - [mlir][AMDGPU] Infer canonical layouts for fat_raw_buffer_cast resetOffset (#149867)

Commit
189 days ago
[mlir][AMDGPU] Infer canonical layouts for fat_raw_buffer_cast resetOffset (#149867) When inferring the return type of amdgpu.fat_raw_buffer_cast with the offset reset, we would sometimes use a strided layout, like strided<[1]>, in cases where, after stripping the offset, the memref had the identity layout. This would cause issues with EmulateNarrowTypes, which does perform this layout canonicalization. Now, the return type inference will put in an identity layout after offset stripping for 1. Statically-shaped memrefs of any rank where the strides match the suffix product of the shape, and 2. Memrefs of rank <= 1 whose strides are [1] (or []) that just had their offset removed by resetOffset.
Author
Parents
Loading