[Mlir-commits] [mlir] [MLIR][XeGPU] Add handling for unit-dim expansion in ShapeCast workgroup-to-subgroup distribution (PR #171758)

[Nishant Patel]

================
@@ -380,4 +380,21 @@ gpu.module @test_1_1_assignment {
     %cst = arith.constant {layout_result_0 = #xegpu.layout<sg_layout = [8, 4], sg_data = [32, 32]>} dense<1.0> : vector<256x128xf32>
     gpu.return
   }
+
+  // CHECK-LABEL: distribute_shapecast_expandunitdims_broadcast
----------------
nbpatel wrote:

please add this test in xegpu-wg-to-sg-unify-ops.mlir file...this file contains old version of nd ops and the plan is delete this test file once we deprecate them

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