Skip to content

Commit d11e4e4

Browse files
committed
Add create_tdesc with sg_map test
1 parent 09376fc commit d11e4e4

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

mlir/test/Dialect/XeGPU/XeGPUOps.mlir

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -131,6 +131,13 @@ gpu.func @test_create_tdesc_vc_1(%src: memref<?xf32, 3>) {
131131
gpu.return
132132
}
133133

134+
// CHECK: gpu.func @test_create_tdesc_vc_with_sg_map(%[[arg0:.*]]: ui64) {
135+
gpu.func @test_create_tdesc_vc_with_sg_map(%src: ui64) {
136+
//CHECK: %[[R0:.*]] = xegpu.create_tdesc %arg0 [0, 8, 16, 24] {chunk_size = 2 : i64} : ui64 -> !xegpu.tensor_desc<4x2xf32, #xegpu.tdesc_attr<scattered = true>, #xegpu.sg_map<wi_layout = [4, 1], wi_data = [1, 1]>>
137+
%1 = xegpu.create_tdesc %src[0, 8, 16, 24] {chunk_size = 2} : ui64 -> !xegpu.tensor_desc<4x2xf32, #xegpu.tdesc_attr<scattered = true>, #xegpu.sg_map<wi_layout = [4, 1], wi_data = [1, 1]>>
138+
gpu.return
139+
}
140+
134141
// CHECK: gpu.func @test_prefetch_vc(%[[arg0:.*]]: ui64) {
135142
gpu.func @test_prefetch_vc(%src: ui64) {
136143
//CHECK: %[[R0:.*]] = xegpu.create_tdesc %arg0 [0, 8, 16, 24] : ui64 -> !xegpu.tensor_desc<4x2xf32, #xegpu.scatter_tdesc_attr<chunk_size = 2 : i64>>

0 commit comments

Comments
 (0)