1// RUN: mlir-opt %s -convert-memref-to-llvm -convert-std-to-llvm -reconcile-unrealized-casts | mlir-cpu-runner -e main -entry-point-result=void -shared-libs=%mlir_runner_utils_dir/libmlir_runner_utils%shlibext,%mlir_runner_utils_dir/libmlir_c_runner_utils%shlibext | FileCheck %s 2 3func private @print_memref_f32(memref<*xf32>) attributes { llvm.emit_c_interface } 4func private @print_memref_i32(memref<*xi32>) attributes { llvm.emit_c_interface } 5func private @printNewline() -> () 6 7memref.global "private" @gv0 : memref<4xf32> = dense<[0.0, 1.0, 2.0, 3.0]> 8func @test1DMemref() { 9 %0 = memref.get_global @gv0 : memref<4xf32> 10 %U = memref.cast %0 : memref<4xf32> to memref<*xf32> 11 // CHECK: rank = 1 12 // CHECK: offset = 0 13 // CHECK: sizes = [4] 14 // CHECK: strides = [1] 15 // CHECK: [0, 1, 2, 3] 16 call @print_memref_f32(%U) : (memref<*xf32>) -> () 17 call @printNewline() : () -> () 18 19 // Overwrite some of the elements. 20 %c0 = constant 0 : index 21 %c2 = constant 2 : index 22 %fp0 = constant 4.0 : f32 23 %fp1 = constant 5.0 : f32 24 memref.store %fp0, %0[%c0] : memref<4xf32> 25 memref.store %fp1, %0[%c2] : memref<4xf32> 26 // CHECK: rank = 1 27 // CHECK: offset = 0 28 // CHECK: sizes = [4] 29 // CHECK: strides = [1] 30 // CHECK: [4, 1, 5, 3] 31 call @print_memref_f32(%U) : (memref<*xf32>) -> () 32 call @printNewline() : () -> () 33 return 34} 35 36memref.global constant @gv1 : memref<3x2xi32> = dense<[[0, 1],[2, 3],[4, 5]]> 37func @testConstantMemref() { 38 %0 = memref.get_global @gv1 : memref<3x2xi32> 39 %U = memref.cast %0 : memref<3x2xi32> to memref<*xi32> 40 // CHECK: rank = 2 41 // CHECK: offset = 0 42 // CHECK: sizes = [3, 2] 43 // CHECK: strides = [2, 1] 44 // CHECK: [0, 1] 45 // CHECK: [2, 3] 46 // CHECK: [4, 5] 47 call @print_memref_i32(%U) : (memref<*xi32>) -> () 48 call @printNewline() : () -> () 49 return 50} 51 52memref.global "private" @gv2 : memref<4x2xf32> = dense<[[0.0, 1.0], [2.0, 3.0], [4.0, 5.0], [6.0, 7.0]]> 53func @test2DMemref() { 54 %0 = memref.get_global @gv2 : memref<4x2xf32> 55 %U = memref.cast %0 : memref<4x2xf32> to memref<*xf32> 56 // CHECK: rank = 2 57 // CHECK: offset = 0 58 // CHECK: sizes = [4, 2] 59 // CHECK: strides = [2, 1] 60 // CHECK: [0, 1] 61 // CHECK: [2, 3] 62 // CHECK: [4, 5] 63 // CHECK: [6, 7] 64 call @print_memref_f32(%U) : (memref<*xf32>) -> () 65 call @printNewline() : () -> () 66 67 // Overwrite the 1.0 (at index [0, 1]) with 10.0 68 %c0 = constant 0 : index 69 %c1 = constant 1 : index 70 %fp10 = constant 10.0 : f32 71 memref.store %fp10, %0[%c0, %c1] : memref<4x2xf32> 72 // CHECK: rank = 2 73 // CHECK: offset = 0 74 // CHECK: sizes = [4, 2] 75 // CHECK: strides = [2, 1] 76 // CHECK: [0, 10] 77 // CHECK: [2, 3] 78 // CHECK: [4, 5] 79 // CHECK: [6, 7] 80 call @print_memref_f32(%U) : (memref<*xf32>) -> () 81 call @printNewline() : () -> () 82 return 83} 84 85memref.global @gv3 : memref<i32> = dense<11> 86func @testScalarMemref() { 87 %0 = memref.get_global @gv3 : memref<i32> 88 %U = memref.cast %0 : memref<i32> to memref<*xi32> 89 // CHECK: rank = 0 90 // CHECK: offset = 0 91 // CHECK: sizes = [] 92 // CHECK: strides = [] 93 // CHECK: [11] 94 call @print_memref_i32(%U) : (memref<*xi32>) -> () 95 call @printNewline() : () -> () 96 return 97} 98 99func @main() -> () { 100 call @test1DMemref() : () -> () 101 call @testConstantMemref() : () -> () 102 call @test2DMemref() : () -> () 103 call @testScalarMemref() : () -> () 104 return 105} 106 107 108