[Mlir-commits] [mlir] [MLIR][XeGPU] Add 2D `vector.multi_reduction` optimization (PR #171154)

[Jianhui Li]

================
@@ -698,9 +706,12 @@ def XeGPU_SliceAttr : XeGPUAttr<"Slice", "slice", [DistributeLayoutAttr]> {
 
     /// Check if this is slice of some other layout.
     bool isSliceOf(const xegpu::DistributeLayoutAttr &other);
-    
+
+    /// Drop the slice dims to get the original layout.
+    xegpu::DistributeLayoutAttr dropSliceDims(ArrayRef<int64_t> dimsToDrop);
----------------
Jianhui-Li wrote:

I think it is better to just make it SliceAttt's interface not DistributeLayoutAttr. Just like flatten().
Return SliceAttr here. 

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