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

Artem Kroviakov llvmlistbot at llvm.org
Thu Jan 30 04:07:12 PST 2025


================
@@ -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]>>
----------------
akroviakov wrote:

The error message is more generic now:
```
Detected a conflict between SG map's work-item layout and TensorDesc shape. Check the index of `subgroup_size` in WI layout map
```

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


More information about the Mlir-commits mailing list