caio/clang-p2996
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[mlir][linalg] Simplify runtime op verification test case (#141454)

Browse Source 
Simplify one of the test cases to make it easier to understand what is
being verified.
This commit is contained in:
Matthias Springer
2025-05-26 14:43:38 +09:00
committed by GitHub
parent 9909cf5303
commit 64f040b985
1 changed files with 4 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
mlir/test/Integration/Dialect/Linalg/CPU/runtime-verification.mlir
View File

@@ -109,15 +109,12 @@ func.func @main() {
#identity1D = affine_map<(d0) -> (d0)>
func.func @simple_add(%arg0: tensor<?xf32>, %arg1: tensor<?xf32>) -> (tensor<?xf32>) {
%c0 = arith.constant 0 : index
%dim = tensor.dim %arg0, %c0 : tensor<?xf32>
%result = tensor.empty(%dim) : tensor<?xf32>
%0 = linalg.generic {
indexing_maps = [#identity1D, #identity1D, #identity1D],
indexing_maps = [#identity1D, #identity1D],
iterator_types = ["parallel"]
} ins(%arg0, %arg1 : tensor<?xf32>, tensor<?xf32>)
outs(%result : tensor<?xf32>) {
^bb0(%gen_arg1: f32, %gen_arg2: f32, %out: f32) :
} ins(%arg0 : tensor<?xf32>)
outs(%arg1 : tensor<?xf32>) {
^bb0(%gen_arg1: f32, %gen_arg2: f32) :
%tmp1 = arith.addf %gen_arg1, %gen_arg2 : f32
linalg.yield %tmp1 : f32
} -> tensor<?xf32>