82 lines · plain
1// RUN: mlir-opt %s -generate-runtime-verification \2// RUN: -expand-strided-metadata \3// RUN: -test-cf-assert \4// RUN: -convert-to-llvm | \5// RUN: mlir-runner -e main -entry-point-result=void \6// RUN: -shared-libs=%mlir_runner_utils 2>&1 | \7// RUN: FileCheck %s8 9// RUN: mlir-opt %s -generate-runtime-verification \10// RUN: -expand-strided-metadata \11// RUN: -test-cf-assert \12// RUN: -convert-to-llvm="allow-pattern-rollback=0" \13// RUN: -reconcile-unrealized-casts | \14// RUN: mlir-runner -e main -entry-point-result=void \15// RUN: -shared-libs=%mlir_runner_utils 2>&1 | \16// RUN: FileCheck %s17 18func.func @cast_to_static_dim(%m: memref<?xf32>) -> memref<10xf32> {19 %0 = memref.cast %m : memref<?xf32> to memref<10xf32>20 return %0 : memref<10xf32>21}22 23func.func @cast_to_ranked(%m: memref<*xf32>) -> memref<f32> {24 %0 = memref.cast %m : memref<*xf32> to memref<f32>25 return %0 : memref<f32>26}27 28func.func @cast_to_static_strides(%m: memref<?xf32, strided<[?], offset: ?>>)29 -> memref<?xf32, strided<[9], offset: 5>> {30 %0 = memref.cast %m : memref<?xf32, strided<[?], offset: ?>>31 to memref<?xf32, strided<[9], offset: 5>>32 return %0 : memref<?xf32, strided<[9], offset: 5>>33}34 35func.func @valid_cast(%m: memref<*xf32>) -> memref<?xf32> {36 %0 = memref.cast %m : memref<*xf32> to memref<?xf32>37 return %0 : memref<?xf32>38}39 40func.func @main() {41 // All casts inside the called functions are invalid at runtime, except for42 // the last one.43 %alloc = memref.alloc() : memref<5xf32>44 45 // CHECK: ERROR: Runtime op verification failed46 // CHECK-NEXT: memref.cast %{{.*}} : memref<?xf32> to memref<10xf32>47 // CHECK-NEXT: ^ size mismatch of dim 048 // CHECK-NEXT: Location: loc({{.*}})49 %1 = memref.cast %alloc : memref<5xf32> to memref<?xf32>50 func.call @cast_to_static_dim(%1) : (memref<?xf32>) -> (memref<10xf32>)51 52 // CHECK-NEXT: ERROR: Runtime op verification failed53 // CHECK-NEXT: memref.cast %{{.*}} : memref<*xf32> to memref<f32>54 // CHECK-NEXT: ^ rank mismatch55 // CHECK-NEXT: Location: loc({{.*}})56 %3 = memref.cast %alloc : memref<5xf32> to memref<*xf32>57 func.call @cast_to_ranked(%3) : (memref<*xf32>) -> (memref<f32>)58 59 // CHECK-NEXT: ERROR: Runtime op verification failed60 // CHECK-NEXT: memref.cast %{{.*}} : memref<?xf32, strided<[?], offset: ?>>61 // CHECK-NEXT: ^ offset mismatch62 // CHECK-NEXT: Location: loc({{.*}})63 64 // CHECK-NEXT: ERROR: Runtime op verification failed65 // CHECK-NEXT: memref.cast %{{.*}} : memref<?xf32, strided<[?], offset: ?>>66 // CHECK-NEXT: ^ stride mismatch of dim 067 // CHECK-NEXT: Location: loc({{.*}})68 %4 = memref.cast %alloc69 : memref<5xf32> to memref<?xf32, strided<[?], offset: ?>>70 func.call @cast_to_static_strides(%4)71 : (memref<?xf32, strided<[?], offset: ?>>)72 -> (memref<?xf32, strided<[9], offset: 5>>)73 74 // A last cast that actually succeeds.75 // CHECK-NOT: ERROR: Runtime op verification failed76 func.call @valid_cast(%3) : (memref<*xf32>) -> (memref<?xf32>)77 78 memref.dealloc %alloc : memref<5xf32>79 80 return81}82