1// RUN: mlir-opt %s -test-vector-to-forloop -convert-vector-to-scf \
2// RUN:   -lower-affine -convert-scf-to-std -convert-vector-to-llvm -convert-std-to-llvm | \
3// RUN: mlir-cpu-runner -e main -entry-point-result=void  \
4// RUN:   -shared-libs=%mlir_integration_test_dir/libmlir_runner_utils%shlibext | \
5// RUN: FileCheck %s
6
7// RUN: mlir-opt %s -convert-vector-to-scf -lower-affine \
8// RUN: -convert-scf-to-std -convert-vector-to-llvm -convert-std-to-llvm | mlir-cpu-runner -e main \
9// RUN: -entry-point-result=void \
10// RUN: -shared-libs=%mlir_integration_test_dir/libmlir_runner_utils%shlibext | \
11// RUN: FileCheck %s
12
13// RUN: mlir-opt %s -test-vector-to-forloop | FileCheck %s -check-prefix=TRANSFORM
14
15
16func private @print_memref_f32(memref<*xf32>)
17
18func @alloc_1d_filled_inc_f32(%arg0: index, %arg1: f32) -> memref<?xf32> {
19  %c0 = constant 0 : index
20  %c1 = constant 1 : index
21  %0 = memref.alloc(%arg0) : memref<?xf32>
22  scf.for %arg2 = %c0 to %arg0 step %c1 {
23    %tmp = index_cast %arg2 : index to i32
24    %tmp1 = sitofp %tmp : i32 to f32
25    %tmp2 = addf %tmp1, %arg1 : f32
26    memref.store %tmp2, %0[%arg2] : memref<?xf32>
27  }
28  return %0 : memref<?xf32>
29}
30
31// Large vector addf that can be broken down into a loop of smaller vector addf.
32func @main() {
33  %cf0 = constant 0.0 : f32
34  %cf1 = constant 1.0 : f32
35  %cf2 = constant 2.0 : f32
36  %c0 = constant 0 : index
37  %c1 = constant 1 : index
38  %c32 = constant 32 : index
39  %c64 = constant 64 : index
40  %out = memref.alloc(%c64) : memref<?xf32>
41  %in1 = call @alloc_1d_filled_inc_f32(%c64, %cf1) : (index, f32) -> memref<?xf32>
42  %in2 = call @alloc_1d_filled_inc_f32(%c64, %cf2) : (index, f32) -> memref<?xf32>
43  // Check that the tansformatio correctly happened.
44  // TRANSFORM: scf.for
45  // TRANSFORM:   vector.transfer_read {{.*}} : memref<?xf32>, vector<2xf32>
46  // TRANSFORM:   vector.transfer_read {{.*}} : memref<?xf32>, vector<2xf32>
47  // TRANSFORM:   %{{.*}} = addf %{{.*}}, %{{.*}} : vector<2xf32>
48  // TRANSFORM:   vector.transfer_write {{.*}} : vector<2xf32>, memref<?xf32>
49  // TRANSFORM: }
50  %a = vector.transfer_read %in1[%c0], %cf0: memref<?xf32>, vector<64xf32>
51  %b = vector.transfer_read %in2[%c0], %cf0: memref<?xf32>, vector<64xf32>
52  %acc = addf %a, %b: vector<64xf32>
53  vector.transfer_write %acc, %out[%c0]: vector<64xf32>, memref<?xf32>
54  %converted = memref.cast %out : memref<?xf32> to memref<*xf32>
55  call @print_memref_f32(%converted): (memref<*xf32>) -> ()
56  // CHECK:      Unranked{{.*}}data =
57  // CHECK:      [
58  // CHECK-SAME:  3,  5,  7,  9,  11,  13,  15,  17,  19,  21,  23,  25,  27,
59  // CHECK-SAME:  29,  31,  33,  35,  37,  39,  41,  43,  45,  47,  49,  51,
60  // CHECK-SAME:  53,  55,  57,  59,  61,  63,  65,  67,  69,  71,  73,  75,
61  // CHECK-SAME:  77,  79,  81,  83,  85,  87,  89,  91,  93,  95,  97,  99,
62  // CHECK-SAME:  101,  103,  105,  107,  109,  111,  113,  115,  117,  119,
63  // CHECK-SAME:  121,  123,  125,  127,  129]
64  memref.dealloc %out : memref<?xf32>
65  memref.dealloc %in1 : memref<?xf32>
66  memref.dealloc %in2 : memref<?xf32>
67  return
68}
69