llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
torch-mlir/test/Dialect/Numpy/ops.mlir

9 lines
352 B
MLIR
Raw Normal View History

Add builtin_ufunc and generic_ufunc ops.
2020-04-29 11:32:49 +08:00
// RUN: npcomp-opt -split-input-file %s | npcomp-opt | FileCheck --dump-input=fail %s
Add !numpy.any_dtype dialect type.
2020-04-30 09:20:42 +08:00
Add builtin_ufunc and generic_ufunc ops.
2020-04-29 11:32:49 +08:00
// -----
// CHECK-LABEL: @builtin_ufunc
Refactor old tracing tests and remove deprecated ops. * Old doctests to run under lit. * Old custom filecheck tests -> pytest directory (under lit). * Rename some old ufunc ops in the tracer.
2020-06-30 07:19:03 +08:00
func @builtin_ufunc(%arg0 : tensor<3xf64>, %arg1 : tensor<3xf64>) -> tensor<3xf64> {
%0 = numpy.builtin_ufunc_call<"numpy.add"> (%arg0, %arg1) : (tensor<3xf64>, tensor<3xf64>) -> tensor<3xf64>
return %0 : tensor<3xf64>
Add numpy.ufunc_call op.
2020-04-30 08:49:56 +08:00
}