Skip to content

Commit ca5c797

Browse files
authored
[TEST] Enable make_range test with a sliced layout (#5427)
1 parent 0b18690 commit ca5c797

File tree

1 file changed

+18
-9
lines changed

1 file changed

+18
-9
lines changed

test/Conversion/tritongpu_to_llvm.mlir

Lines changed: 18 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -387,6 +387,24 @@ module attributes {"ttg.num-ctas" = 1 : i32, "ttg.num-warps" = 1 : i32} {
387387
}
388388
}
389389

390+
391+
// -----
392+
393+
#blocked0 = #ttg.blocked<{sizePerThread = [1, 4], threadsPerWarp = [8, 4], warpsPerCTA = [1, 1], order = [1, 0], CTAsPerCGA = [1, 1], CTASplitNum = [1, 1], CTAOrder = [1, 0]}>
394+
module attributes {"ttg.num-ctas" = 1 : i32, "ttg.num-warps" = 1 : i32} {
395+
// CHECK-LABEL: sliced_layout_make_range
396+
tt.func @sliced_layout_make_range() {
397+
// CHECK: nvvm.read.ptx.sreg.tid.x
398+
// CHECK: llvm.mlir.undef
399+
// CHECK: llvm.insertvalue
400+
// CHECK: llvm.insertvalue
401+
// CHECK: llvm.insertvalue
402+
// CHECK: llvm.insertvalue
403+
%0 = tt.make_range {end = 16 : i32, start = 0 : i32} : tensor<16xi32, #ttg.slice<{dim = 0, parent = #blocked0}>>
404+
tt.return
405+
}
406+
}
407+
390408
// -----
391409

392410
#blocked0 = #ttg.blocked<{sizePerThread = [1], threadsPerWarp = [32], warpsPerCTA = [4], order = [0], CTAsPerCGA = [1], CTASplitNum = [1], CTAOrder = [0]}>
@@ -802,15 +820,6 @@ module attributes {"ttg.num-ctas" = 1 : i32, "ttg.num-warps" = 1 : i32} {
802820
}
803821
}
804822

805-
// TODO: problems in MLIR's parser on slice layout
806-
// #blocked0 = #ttg.blocked<{sizePerThread = [1, 4], threadsPerWarp = [8, 4], warpsPerCTA = [1, 1], order = [1, 0], CTAsPerCGA = [1, 1], CTASplitNum = [1, 1], CTAOrder = [1, 0]}>
807-
// module attributes {"ttg.num-ctas" = 1 : i32, "ttg.num-warps" = 1 : i32} {
808-
// tt.func @make_range_sliced_layout() {
809-
// %0 = tt.make_range {end = 16 : i32, start = 0 : i32} : tensor<16xi32, #ttg.slice<{dim = 0, parent = #blocked0}>>
810-
// tt.return
811-
// }
812-
// }
813-
814823
// -----
815824

816825
#blocked0 = #ttg.blocked<{sizePerThread = [1, 4], threadsPerWarp = [8, 4], warpsPerCTA = [1, 1], order = [1, 0], CTAsPerCGA = [1, 1], CTASplitNum = [1, 1], CTAOrder = [1, 0]}>

0 commit comments

Comments
 (0)