[Mlir-commits] [mlir] [MLIR][XeGPU] Distribute create_nd_desc op without offset from Wg to Sg (PR #152351)
Jianhui Li
llvmlistbot at llvm.org
Fri Aug 8 09:35:10 PDT 2025
================
@@ -7,15 +7,29 @@ gpu.module @test_round_robin_assignment {
// CHECK-COUNT-4: xegpu.create_nd_tdesc %[[ARG_0]][%{{.*}}, %{{.*}}] : memref<256x128xf32>
// CHECK-SAME: -> !xegpu.tensor_desc<16x16xf32, #xegpu.layout<lane_layout = [1, 16], lane_data = [1, 1]>>
// CHECK-NOT: xegpu.create_nd_tdesc
- %tdesc = xegpu.create_nd_tdesc %src[0, 0] : memref<256x128xf32>
+ %cst0 = arith.constant 0 : index
----------------
Jianhui-Li wrote:
I think it is simpler to keep a separate copy of test file, which focus on load_nd/store_nd/prefetch_nd+offset. When transition is over, we simply remove the current test file.
https://github.com/llvm/llvm-project/pull/152351
More information about the Mlir-commits
mailing list