mlir/test/IR/op-stats-json.mlir

   1 // RUN: mlir-opt -allow-unregistered-dialect -print-op-stats=json %s -o=/dev/null 2>&1 | FileCheck %s
   2
   3 func.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: }