intel/llvm
1
0
Fork 0
mirror of https://github.com/intel/llvm.git synced 2026-01-13 19:08:21 +08:00
Code Issues Packages Projects Releases Wiki Activity

[mlir][linalg][arm] Fix use of fill in arm integration tests (#170143)

Browse Source 
Follow up to
https://github.com/llvm/llvm-project/pull/169567#issuecomment-3596220014
This commit is contained in:
Jakub Kuderski
2025-12-01 09:19:07 -05:00
committed by GitHub
parent 2538f6382a
commit ad656d3a19
3 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -22,7 +22,7 @@ func.func @matmul_transpose_a(%A : tensor<?x?xf32>, %B : tensor<?x?xf32>, %C : t
}
func.func @main() {
%c0 = arith.constant 0 : i32
%c0 = arith.constant 0.0 : f32
%c7 = arith.constant 7 : index
%A = arith.constant dense<[
@@ -44,7 +44,7 @@ func.func @main() {
%A_dyn = tensor.cast %A : tensor<13x7xf32> to tensor<?x?xf32>
%C_init = bufferization.alloc_tensor(%c7, %c7) : tensor<?x?xf32>
%C = linalg.fill ins(%c0 : i32) outs(%C_init : tensor<?x?xf32>) -> tensor<?x?xf32>
%C = linalg.fill ins(%c0 : f32) outs(%C_init : tensor<?x?xf32>) -> tensor<?x?xf32>
// CHECK: Unranked Memref {{.*}} rank = 2 offset = 0 sizes = [7, 7] strides = [7, 1] data =
// CHECK: [32955, 33514, 34073, 34632, 35191, 35750, 36309]

4
mlir/test/Integration/Dialect/Linalg/CPU/ArmSME/matmul.mlir
View File

@@ -16,7 +16,7 @@ func.func @matmul(%A : tensor<?x?xf32>, %B : tensor<?x?xf32>, %C : tensor<?x?xf3
}
func.func @main() {
%c0 = arith.constant 0 : i32
%c0 = arith.constant 0.0 : f32
%c7 = arith.constant 7 : index
%A = arith.constant dense<[
@@ -37,7 +37,7 @@ func.func @main() {
%B_dyn = tensor.cast %B : tensor<13x7xf32> to tensor<?x?xf32>
%C_init = bufferization.alloc_tensor(%c7, %c7) : tensor<?x?xf32>
%C = linalg.fill ins(%c0 : i32) outs(%C_init : tensor<?x?xf32>) -> tensor<?x?xf32>
%C = linalg.fill ins(%c0 : f32) outs(%C_init : tensor<?x?xf32>) -> tensor<?x?xf32>
// CHECK: Unranked Memref {{.*}} rank = 2 offset = 0 sizes = [7, 7] strides = [7, 1] data =
// CHECK: [32955, 33514, 34073, 34632, 35191, 35750, 36309]

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

@@ -29,7 +29,7 @@ func.func @main() {
%c128 = arith.constant 128 : i32
func.call @setArmSVLBits(%c128) : (i32) -> ()
%c0 = arith.constant 0 : i32
%c0 = arith.constant 0.0 : f32
%c7 = arith.constant 7 : index
%A = arith.constant dense<[
@@ -50,7 +50,7 @@ func.func @main() {
%B_dyn = tensor.cast %B : tensor<13x7xf32> to tensor<?x?xf32>
%C_init = bufferization.alloc_tensor(%c7, %c7) : tensor<?x?xf32>
%C = linalg.fill ins(%c0 : i32) outs(%C_init : tensor<?x?xf32>) -> tensor<?x?xf32>
%C = linalg.fill ins(%c0 : f32) outs(%C_init : tensor<?x?xf32>) -> tensor<?x?xf32>
// CHECK: Unranked Memref {{.*}} rank = 2 offset = 0 sizes = [7, 7] strides = [7, 1] data =
// CHECK: [32955, 33514, 34073, 34632, 35191, 35750, 36309]