Skip to content

Commit 78d3b27

Browse files
committed
[mlir][linalg][arm] Fix use of fill in arm integration tests
1 parent 3694798 commit 78d3b27

File tree

3 files changed

+6
-6
lines changed

3 files changed

+6
-6
lines changed

mlir/test/Integration/Dialect/Linalg/CPU/ArmSME/matmul-transpose-a.mlir

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,7 @@ func.func @matmul_transpose_a(%A : tensor<?x?xf32>, %B : tensor<?x?xf32>, %C : t
2222
}
2323

2424
func.func @main() {
25-
%c0 = arith.constant 0 : i32
25+
%c0 = arith.constant 0.0 : f32
2626
%c7 = arith.constant 7 : index
2727

2828
%A = arith.constant dense<[
@@ -44,7 +44,7 @@ func.func @main() {
4444
%A_dyn = tensor.cast %A : tensor<13x7xf32> to tensor<?x?xf32>
4545

4646
%C_init = bufferization.alloc_tensor(%c7, %c7) : tensor<?x?xf32>
47-
%C = linalg.fill ins(%c0 : i32) outs(%C_init : tensor<?x?xf32>) -> tensor<?x?xf32>
47+
%C = linalg.fill ins(%c0 : f32) outs(%C_init : tensor<?x?xf32>) -> tensor<?x?xf32>
4848

4949
// CHECK: Unranked Memref {{.*}} rank = 2 offset = 0 sizes = [7, 7] strides = [7, 1] data =
5050
// CHECK: [32955, 33514, 34073, 34632, 35191, 35750, 36309]

mlir/test/Integration/Dialect/Linalg/CPU/ArmSME/matmul.mlir

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@ func.func @matmul(%A : tensor<?x?xf32>, %B : tensor<?x?xf32>, %C : tensor<?x?xf3
1616
}
1717

1818
func.func @main() {
19-
%c0 = arith.constant 0 : i32
19+
%c0 = arith.constant 0.0 : f32
2020
%c7 = arith.constant 7 : index
2121

2222
%A = arith.constant dense<[
@@ -37,7 +37,7 @@ func.func @main() {
3737
%B_dyn = tensor.cast %B : tensor<13x7xf32> to tensor<?x?xf32>
3838

3939
%C_init = bufferization.alloc_tensor(%c7, %c7) : tensor<?x?xf32>
40-
%C = linalg.fill ins(%c0 : i32) outs(%C_init : tensor<?x?xf32>) -> tensor<?x?xf32>
40+
%C = linalg.fill ins(%c0 : f32) outs(%C_init : tensor<?x?xf32>) -> tensor<?x?xf32>
4141

4242
// CHECK: Unranked Memref {{.*}} rank = 2 offset = 0 sizes = [7, 7] strides = [7, 1] data =
4343
// CHECK: [32955, 33514, 34073, 34632, 35191, 35750, 36309]

mlir/test/Integration/Dialect/Linalg/CPU/ArmSME/multi-tile-matmul.mlir

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -29,7 +29,7 @@ func.func @main() {
2929
%c128 = arith.constant 128 : i32
3030
func.call @setArmSVLBits(%c128) : (i32) -> ()
3131

32-
%c0 = arith.constant 0 : i32
32+
%c0 = arith.constant 0.0 : f32
3333
%c7 = arith.constant 7 : index
3434

3535
%A = arith.constant dense<[
@@ -50,7 +50,7 @@ func.func @main() {
5050
%B_dyn = tensor.cast %B : tensor<13x7xf32> to tensor<?x?xf32>
5151

5252
%C_init = bufferization.alloc_tensor(%c7, %c7) : tensor<?x?xf32>
53-
%C = linalg.fill ins(%c0 : i32) outs(%C_init : tensor<?x?xf32>) -> tensor<?x?xf32>
53+
%C = linalg.fill ins(%c0 : f32) outs(%C_init : tensor<?x?xf32>) -> tensor<?x?xf32>
5454

5555
// CHECK: Unranked Memref {{.*}} rank = 2 offset = 0 sizes = [7, 7] strides = [7, 1] data =
5656
// CHECK: [32955, 33514, 34073, 34632, 35191, 35750, 36309]

0 commit comments

Comments
 (0)