| // 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 |
| %f5 = constant 5.00000e+00 : f32 |
| %f10 = constant 10.00000e+00 : f32 |
| |
| %V = memref_cast %A : memref<10x3xf32, 0> to memref<?x?xf32> |
| linalg.fill(%V, %f10) : memref<?x?xf32, 0>, f32 |
| %U = memref_cast %A : memref<10x3xf32, 0> to memref<*xf32> |
| call @print_memref_f32(%U) : (memref<*xf32>) -> () |
| |
| %V2 = memref_cast %U : memref<*xf32> to memref<?x?xf32> |
| linalg.fill(%V2, %f5) : memref<?x?xf32, 0>, f32 |
| %U2 = memref_cast %V2 : memref<?x?xf32, 0> to memref<*xf32> |
| call @print_memref_f32(%U2) : (memref<*xf32>) -> () |
| |
| %V3 = memref_cast %V2 : memref<?x?xf32> to memref<*xf32> |
| %V4 = memref_cast %V3 : memref<*xf32> to memref<?x?xf32> |
| linalg.fill(%V4, %f2) : memref<?x?xf32, 0>, f32 |
| %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>) |