37 lines
2 KiB
MLIR
37 lines
2 KiB
MLIR
// RUN: mlir-opt -allow-unregistered-dialect -print-op-stats=json %s -o=/dev/null 2>&1 | FileCheck %s
|
|
|
|
func.func @main(tensor<4xf32>, tensor<4xf32>) -> tensor<4xf32> {
|
|
^bb0(%arg0: tensor<4xf32>, %arg1: tensor<4xf32>):
|
|
%0 = arith.addf %arg0, %arg1 : tensor<4xf32>
|
|
%1 = arith.addf %arg0, %arg1 : tensor<4xf32>
|
|
%2 = arith.addf %arg0, %arg1 : tensor<4xf32>
|
|
%3 = arith.addf %arg0, %arg1 : tensor<4xf32>
|
|
%4 = arith.addf %arg0, %arg1 : tensor<4xf32>
|
|
%5 = arith.addf %arg0, %arg1 : tensor<4xf32>
|
|
%10 = "xla.add"(%0, %arg1) : (tensor<4xf32>,tensor<4xf32>)-> tensor<4xf32>
|
|
%11 = "xla.add"(%0, %arg1) : (tensor<4xf32>,tensor<4xf32>)-> tensor<4xf32>
|
|
%12 = "xla.add"(%0, %arg1) : (tensor<4xf32>,tensor<4xf32>)-> tensor<4xf32>
|
|
%13 = "xla.add"(%0, %arg1) : (tensor<4xf32>,tensor<4xf32>)-> tensor<4xf32>
|
|
%14 = "xla.add"(%0, %arg1) : (tensor<4xf32>,tensor<4xf32>)-> tensor<4xf32>
|
|
%15 = "xla.add"(%0, %arg1) : (tensor<4xf32>,tensor<4xf32>)-> tensor<4xf32>
|
|
%16 = "xla.add"(%0, %arg1) : (tensor<4xf32>,tensor<4xf32>)-> tensor<4xf32>
|
|
%17 = "xla.add"(%0, %arg1) : (tensor<4xf32>,tensor<4xf32>)-> tensor<4xf32>
|
|
%18 = "xla.add"(%0, %arg1) : (tensor<4xf32>,tensor<4xf32>)-> tensor<4xf32>
|
|
%19 = "xla.add"(%0, %arg1) : (tensor<4xf32>,tensor<4xf32>)-> tensor<4xf32>
|
|
%20 = "xla.add"(%0, %arg1) : (tensor<4xf32>,tensor<4xf32>)-> tensor<4xf32>
|
|
%21 = "xla.add"(%0, %arg1) : (tensor<4xf32>,tensor<4xf32>)-> tensor<4xf32>
|
|
%22 = "xla.add"(%0, %arg1) : (tensor<4xf32>,tensor<4xf32>)-> tensor<4xf32>
|
|
%23 = "xla.add"(%0, %arg1) : (tensor<4xf32>,tensor<4xf32>)-> tensor<4xf32>
|
|
%24 = "xla.add"(%0, %arg1) : (tensor<4xf32>,tensor<4xf32>)-> tensor<4xf32>
|
|
%25 = "xla.add"(%0, %arg1) : (tensor<4xf32>,tensor<4xf32>)-> tensor<4xf32>
|
|
%26 = "xla.add"(%0, %arg1) : (tensor<4xf32>,tensor<4xf32>)-> tensor<4xf32>
|
|
%30 = "long_op_name"(%0, %arg1) : (tensor<4xf32>,tensor<4xf32>)-> tensor<4xf32>
|
|
return %1 : tensor<4xf32>
|
|
}
|
|
|
|
// CHECK: {
|
|
// CHECK: "arith.addf" : 6,
|
|
// CHECK: "func.return" : 1,
|
|
// CHECK: "long_op_name" : 1,
|
|
// CHECK: "xla.add" : 17
|
|
// CHECK: }
|