[Mlir-commits] [mlir] [mlir][xegpu] Bug fix in UpdateNdOffset distribution. (PR #150545)

[Adam Siemieniuk]

================
@@ -265,6 +265,28 @@ gpu.module @test {
   }
 }
 
+// -----
+// Explicitly check that update_nd_offset distributed tensor descriptor retains the layouts.
----------------
adam-smnk wrote:

nit: `update_nd_offset` doesn't really retain layouts, the problem is more mechanical when remaining `gpu.warp_execute_on_lane_0` return type is different than what the newly generated consumer expects

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