61 lines · plain
1// RUN: mlir-opt %s -convert-vector-to-scf -expand-realloc -expand-strided-metadata -convert-scf-to-cf -convert-vector-to-llvm -finalize-memref-to-llvm -convert-func-to-llvm -convert-arith-to-llvm -convert-cf-to-llvm -reconcile-unrealized-casts |\2// RUN: mlir-runner -e entry -entry-point-result=void \3// RUN: -shared-libs=%mlir_c_runner_utils4// RUN: mlir-opt %s -convert-vector-to-scf -expand-realloc -expand-strided-metadata -convert-scf-to-cf -convert-vector-to-llvm -finalize-memref-to-llvm='use-aligned-alloc=1' -convert-func-to-llvm -arith-expand -convert-arith-to-llvm -convert-cf-to-llvm -reconcile-unrealized-casts |\5// RUN: mlir-runner -e entry -entry-point-result=void \6// RUN: -shared-libs=%mlir_c_runner_utils | FileCheck %s7 8// FIXME: Windows does not have aligned_alloc9// UNSUPPORTED: system-windows10 11func.func @entry() {12 // Set up memory.13 %c0 = arith.constant 0: index14 %c1 = arith.constant 1: index15 %c8 = arith.constant 8: index16 %A = memref.alloc() : memref<8xf32>17 scf.for %i = %c0 to %c8 step %c1 {18 %i32 = arith.index_cast %i : index to i3219 %fi = arith.sitofp %i32 : i32 to f3220 memref.store %fi, %A[%i] : memref<8xf32>21 }22 23 %d0 = arith.constant -1.0 : f3224 %Av = vector.transfer_read %A[%c0], %d0: memref<8xf32>, vector<8xf32>25 vector.print %Av : vector<8xf32>26 // CHECK: ( 0, 1, 2, 3, 4, 5, 6, 7 )27 28 // Realloc with static sizes.29 %B = memref.realloc %A : memref<8xf32> to memref<10xf32>30 31 %c10 = arith.constant 10: index32 scf.for %i = %c8 to %c10 step %c1 {33 %i32 = arith.index_cast %i : index to i3234 %fi = arith.sitofp %i32 : i32 to f3235 memref.store %fi, %B[%i] : memref<10xf32>36 }37 38 %Bv = vector.transfer_read %B[%c0], %d0: memref<10xf32>, vector<10xf32>39 vector.print %Bv : vector<10xf32>40 // CHECK: ( 0, 1, 2, 3, 4, 5, 6, 7, 8, 9 )41 42 // Realloc with dynamic sizes.43 %Bd = memref.cast %B : memref<10xf32> to memref<?xf32>44 %c13 = arith.constant 13: index45 %Cd = memref.realloc %Bd(%c13) : memref<?xf32> to memref<?xf32>46 %C = memref.cast %Cd : memref<?xf32> to memref<13xf32>47 48 scf.for %i = %c10 to %c13 step %c1 {49 %i32 = arith.index_cast %i : index to i3250 %fi = arith.sitofp %i32 : i32 to f3251 memref.store %fi, %C[%i] : memref<13xf32>52 }53 54 %Cv = vector.transfer_read %C[%c0], %d0: memref<13xf32>, vector<13xf32>55 vector.print %Cv : vector<13xf32>56 // CHECK: ( 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12 )57 58 memref.dealloc %C : memref<13xf32>59 return60}61