Skip to content

Commit 3b389bf

Browse files
committed
add 1D layout examples
1 parent 9ed0f87 commit 3b389bf

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

mlir/include/mlir/Dialect/XeGPU/IR/XeGPUTypes.td

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -77,6 +77,12 @@ def XeGPU_TensorDesc: XeGPUTypeDef<"TensorDesc", "tensor_desc",
7777
// A TensorDesc with 8x16 f32 elements for a memory region in shared memory space.
7878
xegpu.tensor_desc<8x16xf32, #xegpu.tdesc_attr<memory_space = slm>>
7979

80+
// A 1D TensorDesc with a layout for subgroup level programming, each lane access two continuous elements
81+
xegpu.tensor_desc<32xf32, #xegpu.layout<lane_layout = [16], lane_data = [2]>>
82+
83+
// A 1D TensorDesc with a layout for subgroup level programming, each lane access two elements with stride = 16
84+
xegpu.tensor_desc<32xf32, #xegpu.layout<lane_layout = [16], lane_data = [1]>>
85+
8086
// A TensorDesc with a layout for subgroup level programming
8187
xegpu.tensor_desc<8x16xf32, #xegpu.layout<lane_layout = [1, 16], lane_data = [1, 1]>>
8288

0 commit comments

Comments
 (0)