[Mlir-commits] [mlir] [MLIR][XeGPU] Fold lane_data repack convert_layout (PR #204016)

[Artem Kroviakov]

================
@@ -1638,6 +1638,45 @@ struct SgToLaneConvertLayout
       }
     }
 
+    // Special case: fold a convert_layout that merely repacks `lane_data`
+    // along the non-distributed outer dimension - from `[N, 1]` to `[1, 1]`
+    // with `order = [1, 0]`, keeping `lane_layout` unchanged - when its result
+    // is consumed by exactly `N` `vector.extract_strided_slice` ops. After lane
----------------
akroviakov wrote:

> when its result is consumed by exactly `N` `vector.extract_strided_slice` ops.

Can our pipeline produce sg-level code where this is not the case?

https://github.com/llvm/llvm-project/pull/204016