[Mlir-commits] [mlir] [XeGPU] Add sg_map for scatter verification (PR #124300)

[Adam Siemieniuk]

================
@@ -170,6 +170,64 @@ func.func @test_prefetch_vc_2(%src: ui64) {
   return
 }
 
+// -----
+func.func @test_create_tdesc_sg_map_1(%src: ui64) {
+  %cst = arith.constant dense<[0, 8, 16, 24]> : vector<4xindex>
+  // expected-error at +1 {{TensorDesc of a 1D tensor must have wi_layout[1] == tdescShape[0]}}
+  %1 = xegpu.create_tdesc %src, %cst : ui64, vector<4xindex> -> !xegpu.tensor_desc<4xf32, #xegpu.scatter_tdesc_attr<>, #xegpu.sg_map<wi_layout = [4, 1], wi_data = [1, 1]>>
----------------
adam-smnk wrote:

I'm not sure that's exactly correct (judging by the error message, maybe the login check is fine).
For example, I think such descriptor should be fine:
`!xegpu.tensor_desc<2xf32, #xegpu.scatter_tdesc_attr<chunk_size = 2>, #xegpu.sg_map<wi_layout = [1, 1], wi_data = [1, 2]>>`

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