llvm-optiler/trace.log at main · TiElixir/llvm-optiler · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
module {
  func.func @test(
      %arg0: tensor<4x4xf32>,
      %arg1: tensor<4x4xf32>,
      %arg2: tensor<4xf32>
  ) -> tensor<4x4xf32> {

    %cst = arith.constant 0.0 : f32
    %0 = tensor.empty() : tensor<4x4xf32>

    %1 = linalg.fill ins(%cst : f32) outs(%0 : tensor<4x4xf32>)
      -> tensor<4x4xf32>

    %2 = linalg.matmul
      ins(%arg0, %arg1 : tensor<4x4xf32>, tensor<4x4xf32>)
      outs(%1 : tensor<4x4xf32>)
      -> tensor<4x4xf32>

    %3 = linalg.generic
      ins(%1, %2 : tensor<4x4xf32>, tensor<4x4xf32>)
      outs(%0 : tensor<4x4xf32>)
      {
        ^bb0(%in: f32, %in_0: f32, %out: f32):
          %4 = arith.addf %in, %in_0 : f32
          linalg.yield %4 : f32
      }
      -> tensor<4x4xf32>

    return %3 : tensor<4x4xf32>
  }
}