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_utils 4// 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 %s 7 8// FIXME: Windows does not have aligned_alloc 9// UNSUPPORTED: system-windows 10 11func.func @entry() { 12 // Set up memory. 13 %c0 = arith.constant 0: index 14 %c1 = arith.constant 1: index 15 %c8 = arith.constant 8: index 16 %A = memref.alloc() : memref<8xf32> 17 scf.for %i = %c0 to %c8 step %c1 { 18 %i32 = arith.index_cast %i : index to i32 19 %fi = arith.sitofp %i32 : i32 to f32 20 memref.store %fi, %A[%i] : memref<8xf32> 21 } 22 23 %d0 = arith.constant -1.0 : f32 24 %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: index 32 scf.for %i = %c8 to %c10 step %c1 { 33 %i32 = arith.index_cast %i : index to i32 34 %fi = arith.sitofp %i32 : i32 to f32 35 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: index 45 %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 i32 50 %fi = arith.sitofp %i32 : i32 to f32 51 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 return 60} 61