xref: /llvm-project/mlir/test/IR/op-stats-json.mlir (revision 8010d7e0446a09471d5349466b80651c3ad76af3)
1// RUN: mlir-opt -allow-unregistered-dialect -print-op-stats=json %s -o=/dev/null 2>&1 | FileCheck %s
2
3func.func @main(tensor<4xf32>, tensor<4xf32>) -> tensor<4xf32> {
4^bb0(%arg0: tensor<4xf32>, %arg1: tensor<4xf32>):
5  %0 = arith.addf %arg0, %arg1 : tensor<4xf32>
6  %1 = arith.addf %arg0, %arg1 : tensor<4xf32>
7  %2 = arith.addf %arg0, %arg1 : tensor<4xf32>
8  %3 = arith.addf %arg0, %arg1 : tensor<4xf32>
9  %4 = arith.addf %arg0, %arg1 : tensor<4xf32>
10  %5 = arith.addf %arg0, %arg1 : tensor<4xf32>
11  %10 = "xla.add"(%0, %arg1) : (tensor<4xf32>,tensor<4xf32>)-> tensor<4xf32>
12  %11 = "xla.add"(%0, %arg1) : (tensor<4xf32>,tensor<4xf32>)-> tensor<4xf32>
13  %12 = "xla.add"(%0, %arg1) : (tensor<4xf32>,tensor<4xf32>)-> tensor<4xf32>
14  %13 = "xla.add"(%0, %arg1) : (tensor<4xf32>,tensor<4xf32>)-> tensor<4xf32>
15  %14 = "xla.add"(%0, %arg1) : (tensor<4xf32>,tensor<4xf32>)-> tensor<4xf32>
16  %15 = "xla.add"(%0, %arg1) : (tensor<4xf32>,tensor<4xf32>)-> tensor<4xf32>
17  %16 = "xla.add"(%0, %arg1) : (tensor<4xf32>,tensor<4xf32>)-> tensor<4xf32>
18  %17 = "xla.add"(%0, %arg1) : (tensor<4xf32>,tensor<4xf32>)-> tensor<4xf32>
19  %18 = "xla.add"(%0, %arg1) : (tensor<4xf32>,tensor<4xf32>)-> tensor<4xf32>
20  %19 = "xla.add"(%0, %arg1) : (tensor<4xf32>,tensor<4xf32>)-> tensor<4xf32>
21  %20 = "xla.add"(%0, %arg1) : (tensor<4xf32>,tensor<4xf32>)-> tensor<4xf32>
22  %21 = "xla.add"(%0, %arg1) : (tensor<4xf32>,tensor<4xf32>)-> tensor<4xf32>
23  %22 = "xla.add"(%0, %arg1) : (tensor<4xf32>,tensor<4xf32>)-> tensor<4xf32>
24  %23 = "xla.add"(%0, %arg1) : (tensor<4xf32>,tensor<4xf32>)-> tensor<4xf32>
25  %24 = "xla.add"(%0, %arg1) : (tensor<4xf32>,tensor<4xf32>)-> tensor<4xf32>
26  %25 = "xla.add"(%0, %arg1) : (tensor<4xf32>,tensor<4xf32>)-> tensor<4xf32>
27  %26 = "xla.add"(%0, %arg1) : (tensor<4xf32>,tensor<4xf32>)-> tensor<4xf32>
28  %30 = "long_op_name"(%0, %arg1) : (tensor<4xf32>,tensor<4xf32>)-> tensor<4xf32>
29  return %1 : tensor<4xf32>
30}
31
32// CHECK: {
33// CHECK:   "arith.addf" : 6,
34// CHECK:   "func.return" : 1,
35// CHECK:   "long_op_name" : 1,
36// CHECK:   "xla.add" : 17
37// CHECK: }
38