Skip to content

Commit 4779124

Browse files
committed
[Intel] Update lit tests after 'e93fc76'
Signed-off-by: Anatoly Myachev <[email protected]>
1 parent fbe22ae commit 4779124

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

test/TritonIntelGPU/combine.mlir

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2925,7 +2925,7 @@ tt.func @hoist_multiple_conditional(
29252925
}
29262926
%2 = arith.addf %0, %1 : tensor<128x32xf32, #blocked>
29272927
%3 = ttg.convert_layout %2 : tensor<128x32xf32, #blocked> -> tensor<128x32xf32, #ttg.dot_op<{opIdx = 0, parent = #mma, kWidth = 2}>>
2928-
%4 = tt.dot %3, %arg4, %arg5 : tensor<128x32xf32, #ttg.dot_op<{opIdx = 0, parent = #mma, kWidth = 2}>> * tensor<32x128xf32, #ttg.dot_op<{opIdx = 1, parent = #mma, kWidth = 2}>> -> tensor<128x128xf32, #mma>
2928+
%4 = tt.dot %3, %arg4, %arg5, inputPrecision = tf32 : tensor<128x32xf32, #ttg.dot_op<{opIdx = 0, parent = #mma, kWidth = 2}>> * tensor<32x128xf32, #ttg.dot_op<{opIdx = 1, parent = #mma, kWidth = 2}>> -> tensor<128x128xf32, #mma>
29292929
tt.return %4 : tensor<128x128xf32, #mma>
29302930
}
29312931

@@ -2954,7 +2954,7 @@ tt.func @hoist_across_loop(
29542954
}
29552955
// CHECK-NOT: ttg.convert_layout
29562956
%2 = ttg.convert_layout %1 : tensor<128x32xf32, #blocked> -> tensor<128x32xf32, #ttg.dot_op<{opIdx = 0, parent = #mma, kWidth = 2}>>
2957-
%3 = tt.dot %2, %arg2, %acc : tensor<128x32xf32, #ttg.dot_op<{opIdx = 0, parent = #mma, kWidth = 2}>> * tensor<32x128xf32, #ttg.dot_op<{opIdx = 1, parent = #mma, kWidth = 2}>> -> tensor<128x128xf32, #mma>
2957+
%3 = tt.dot %2, %arg2, %acc, inputPrecision = tf32 : tensor<128x32xf32, #ttg.dot_op<{opIdx = 0, parent = #mma, kWidth = 2}>> * tensor<32x128xf32, #ttg.dot_op<{opIdx = 1, parent = #mma, kWidth = 2}>> -> tensor<128x128xf32, #mma>
29582958
scf.yield %1, %3 : tensor<128x32xf32, #blocked>, tensor<128x128xf32, #mma>
29592959
}
29602960
tt.return %0#1 : tensor<128x128xf32, #mma>

0 commit comments

Comments
 (0)