torch-mlir/test/Dialect/Numpy/ops.mlir

10 lines
220 B
MLIR
Raw Normal View History

2020-04-27 08:55:15 +08:00
// RUN: npcomp-opt %s | npcomp-opt | FileCheck %s
// CHECK-LABEL: func @foo()
func @foo() -> i32 {
%0 = constant 1 : i32
// CHECK: %{{.*}} = numpy.foo %{{.*}} : i32
%res = numpy.foo %0 : i32
return %res : i32
}