diff options
Diffstat (limited to 'mlir/test/mlir-cpu-runner/unranked_memref.mlir')
-rw-r--r-- | mlir/test/mlir-cpu-runner/unranked_memref.mlir | 16 |
1 files changed, 16 insertions, 0 deletions
diff --git a/mlir/test/mlir-cpu-runner/unranked_memref.mlir b/mlir/test/mlir-cpu-runner/unranked_memref.mlir index 4e721be..7447e9d 100644 --- a/mlir/test/mlir-cpu-runner/unranked_memref.mlir +++ b/mlir/test/mlir-cpu-runner/unranked_memref.mlir @@ -1,19 +1,27 @@ // RUN: mlir-opt %s -convert-linalg-to-loops -convert-linalg-to-llvm -convert-std-to-llvm | mlir-cpu-runner -e main -entry-point-result=void -shared-libs=%linalg_test_lib_dir/libmlir_runner_utils%shlibext,%linalg_test_lib_dir/libcblas%shlibext,%linalg_test_lib_dir/libcblas_interface%shlibext | FileCheck %s // CHECK: rank = 2 +// CHECK: rank = 2 // CHECK-SAME: sizes = [10, 3] // CHECK-SAME: strides = [3, 1] // CHECK-COUNT-10: [10, 10, 10] // // CHECK: rank = 2 +// CHECK: rank = 2 // CHECK-SAME: sizes = [10, 3] // CHECK-SAME: strides = [3, 1] // CHECK-COUNT-10: [5, 5, 5] // // CHECK: rank = 2 +// CHECK: rank = 2 // CHECK-SAME: sizes = [10, 3] // CHECK-SAME: strides = [3, 1] // CHECK-COUNT-10: [2, 2, 2] +// +// CHECK: rank = 0 +// CHECK: rank = 0 +// 122 is ASCII for 'z'. +// CHECK: [z] func @main() -> () { %A = alloc() : memref<10x3xf32, 0> %f2 = constant 2.00000e+00 : f32 @@ -36,8 +44,16 @@ func @main() -> () { %U3 = memref_cast %V2 : memref<?x?xf32> to memref<*xf32> call @print_memref_f32(%U3) : (memref<*xf32>) -> () + // 122 is ASCII for 'z'. + %i8_z = constant 122 : i8 + %I8 = alloc() : memref<i8> + store %i8_z, %I8[]: memref<i8> + %U4 = memref_cast %I8 : memref<i8> to memref<*xi8> + call @print_memref_i8(%U4) : (memref<*xi8>) -> () + dealloc %A : memref<10x3xf32, 0> return } +func @print_memref_i8(memref<*xi8>) func @print_memref_f32(memref<*xf32>) |