@@ -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 <16 xi32 , #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