diff --git a/test/TritonGPU/consan.mlir b/test/TritonGPU/consan.mlir index 7d908a9e2e56..9ef0928b574c 100644 --- a/test/TritonGPU/consan.mlir +++ b/test/TritonGPU/consan.mlir @@ -264,7 +264,7 @@ module attributes {"ttg.num-ctas" = 2 : i32, "ttg.num-warps" = 1 : i32, ttg.shar // CHECK: arith.shrui // CHECK-LABEL: @outstanding_commits_multicast_tma_recipients tt.func public @outstanding_commits_multicast_tma_recipients( - %desc: !tt.tensordesc>, + %desc: !tt.tensordesc<32x32xf32, #shared>, %ptr: tensor<32x32x!tt.ptr, #blocked>) { %true = arith.constant true %c0_i32 = arith.constant 0 : i32 @@ -286,7 +286,7 @@ module attributes {"ttg.num-ctas" = 2 : i32, "ttg.num-warps" = 1 : i32, ttg.shar // CHECK: %[[RECIPIENTS:.*]] = arith.shli %[[PATTERN]], // CHECK: tt.call @__triton_consan_check_outstanding_commits{{.*}}({{.*}}, %[[RECIPIENTS]]) // CHECK: ttng.async_tma_copy_global_to_local - ttng.async_tma_copy_global_to_local %desc[%c0_i32, %c0_i32] %shmem, %bar, %true {multicast} : !tt.tensordesc>, !ttg.memdesc<2xi64, #shared1, #smem, mutable> -> !ttg.memdesc<32x32xf32, #shared, #smem, mutable> + ttng.async_tma_copy_global_to_local %desc[%c0_i32, %c0_i32] %shmem, %bar, %true {multicast} : !tt.tensordesc<32x32xf32, #shared>, !ttg.memdesc<2xi64, #shared1, #smem, mutable> -> !ttg.memdesc<32x32xf32, #shared, #smem, mutable> tt.return } }