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