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

Add test case for !numpy.ndarray.

pull/1/head
Stella Laurenzo 2020-06-28 17:41:21 -07:00
parent efe8915901
commit b2708e4687
1 changed files with 13 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
test/Dialect/Numpy/types_attrs.mlir 100644
View File

@ -0,0 +1,13 @@
// RUN: npcomp-opt -split-input-file %s | npcomp-opt | FileCheck --dump-input=fail %s
// CHECK-LABEL: @ndarray_no_dtype
// CHECK: !numpy.ndarray<?>
func @ndarray_no_dtype(%arg0 : !numpy.ndarray<?>) -> !numpy.ndarray<?> {
return %arg0 : !numpy.ndarray<?>
}
// CHECK-LABEL: @ndarray_dtype
// CHECK: !numpy.ndarray<i32>
func @ndarray_dtype(%arg0 : !numpy.ndarray<i32>) -> !numpy.ndarray<i32> {
return %arg0 : !numpy.ndarray<i32>
}