mirror of https://github.com/llvm/torch-mlir
16 lines
732 B
MLIR
16 lines
732 B
MLIR
// RUN: npcomp-opt %s -aten-layer-name -aten-op-report |& FileCheck %s
|
|
// CHECK-LABEL: "L0-add-0": {
|
|
// CHECK-NEXT: "activation_in": 12,
|
|
// CHECK-NEXT: "activation_out": 6,
|
|
// CHECK-NEXT: "ops:+": 6,
|
|
// CHECK-NEXT: "reads": 12,
|
|
// CHECK-NEXT: "writes": 6
|
|
|
|
// RUN: npcomp-opt %s -aten-to-std |& FileCheck %s --check-prefix=CHECK-CONVERSION
|
|
// CHECK-CONVERSION-LABEL: @graph
|
|
func @graph(%arg0: tensor<1x2x3xf32>, %arg1: tensor<1x2x3xf32>) -> tensor<1x2x3xf32> {
|
|
%1 = "aten.constant"() {type = "i32", value = 1 : i32} : () -> i32
|
|
%2 = "aten.add"(%arg0, %arg1, %1) : (tensor<1x2x3xf32>, tensor<1x2x3xf32>, i32) -> tensor<1x2x3xf32>
|
|
"std.return"(%2) : (tensor<1x2x3xf32>) -> ()
|
|
}
|