[Mlir-commits] [mlir] [MLIR][XeGPU] Allow create mem desc from 2d memref (PR #167767)

[Jianhui Li]

================
@@ -1304,11 +1304,11 @@ def XeGPU_CreateMemDescOp: XeGPU_Op<"create_mem_desc", [Pure,
     as the underlying shared local memory.
 
     Arguments:
-     - `source` : a 1D statically shaped memref with element type i8, representing the raw SLM buffer.
+     - `source` : 1D or 2D statically shape memref, representing the raw SLM buffer. When the source is provided as 1D memref, its type must be i8. 
----------------
Jianhui-Li wrote:

I relaxed it to XeGPU_ScalarType also

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