[Mlir-commits] [mlir] [mlir][xegpu] Remove chunk_size attribute; infer from types (PR #205122)

Jianhui Li llvmlistbot at llvm.org
Wed Sep 2 15:58:28 PDT 2026


================
@@ -708,15 +708,12 @@ def XeGPU_LoadGatherOp : XeGPU_Op<"load", [MemoryEffects<[MemRead]>, AnchorLayou
 
   let description = [{ It (aka. load) load data per each lane. The output
     describes the data being loaded at the subgroup level, so its size is
-    consistent with the number of lanes in a subgroup. When the chunk size
-    is larger than 2, the output vector is a 2D vector, with dim-0 correspoding
-    to lanes, and dim-1 corresponding to the chunk size loaded by each lane.
+    consistent with the number of lanes in a subgroup. At lane level each lane
+    may load a 1D vector of contiguous elements; that per-lane width is implied
+    by the result type and does not need a separate attribute.
----------------
Jianhui-Li wrote:

Please add:
For each level, the mask, offsets, and result can be nD vectors, and their shape must be same. 

I don't think we need to describe this "each lane may load 1d vector" in the description.  The lane level could load nd vector with leading unit dims. 


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


More information about the Mlir-commits mailing list