Skip to content

N
ngraph
Commit 92d24fd0 authored by Diego Caballero's avatar Diego Caballero Committed by Scott Cyphers
Browse files
Options

[MLIR] Minor improvement to elementwise LIT test (#3666)

parent 80022ea1
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 1 deletion
test/mlir/ngraph_dialect/elementwise_binary_ops.mlir
View file @ 92d24fd0
// RUN: ngraph-opt %s | FileCheck %s // RUN: ngraph-opt %s | FileCheck %s
// Verify the printed output can be parsed.
// RUN: ngraph-opt %s | ngraph-opt | FileCheck %s
// These tests verify the parser, builder and printer of element-wise binary ops. // These tests verify the parser, builder and printer of element-wise binary ops.
// CHECK-LABEL: func @add_float // CHECK-LABEL: func @add_float
func @add_float(%arg0: !ng.tensor<2x2xf32>, %arg1: !ng.tensor<2x2xf32>) -> !ng.tensor<2x2xf32> { func @add_float(%arg0: !ng.tensor<2x2xf32>, %arg1: !ng.tensor<2x2xf32>) -> !ng.tensor<2x2xf32> {
// CHECK: ng.add // CHECK: %{{[0-9]+}} = "ng.add"(%{{.*}}, %{{.*}}) : (!ng.tensor<2x2xf32>, !ng.tensor<2x2xf32>) -> !ng.tensor<2x2xf32>
%0 = "ng.add"(%arg1, %arg0) : (!ng.tensor<2x2xf32>, !ng.tensor<2x2xf32>) -> !ng.tensor<2x2xf32> %0 = "ng.add"(%arg1, %arg0) : (!ng.tensor<2x2xf32>, !ng.tensor<2x2xf32>) -> !ng.tensor<2x2xf32>
"ng.return"(%0) : (!ng.tensor<2x2xf32>) -> () "ng.return"(%0) : (!ng.tensor<2x2xf32>) -> ()
} }
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment