| // RUN: mlir-query %s -c "m getDefinitionsByPredicate(hasOpName(\"memref.store\"),hasOpName(\"memref.alloc\"),true,false,false).extract(\"backward_slice\")" | FileCheck %s | |
| // CHECK: func.func @backward_slice(%{{.*}}: memref<10xf32>) -> (f32, index, index, f32, index, index, f32) { | |
| // CHECK: %[[CST0:.*]] = arith.constant 0.000000e+00 : f32 | |
| // CHECK-NEXT: %[[C0:.*]] = arith.constant 0 : index | |
| // CHECK-NEXT: %[[I0:.*]] = affine.apply affine_map<()[s0] -> (s0)>()[%[[C0]]] | |
| // CHECK-NEXT: memref.store %[[CST0]], %{{.*}}[%[[I0]]] : memref<10xf32> | |
| // CHECK-NEXT: %[[CST2:.*]] = arith.constant 0.000000e+00 : f32 | |
| // CHECK-NEXT: %[[I1:.*]] = affine.apply affine_map<() -> (0)>() | |
| // CHECK-NEXT: memref.store %[[CST2]], %{{.*}}[%[[I1]]] : memref<10xf32> | |
| // CHECK-NEXT: %[[C1:.*]] = arith.constant 0 : index | |
| // CHECK-NEXT: %[[LOAD:.*]] = memref.load %{{.*}}[%[[C1]]] : memref<10xf32> | |
| // CHECK-NEXT: memref.store %[[LOAD]], %{{.*}}[%[[C1]]] : memref<10xf32> | |
| // CHECK-NEXT: return %[[CST0]], %[[C0]], %[[I0]], %[[CST2]], %[[I1]], %[[C1]], %[[LOAD]] : f32, index, index, f32, index, index, f32 | |
| func.func @slicing_memref_store_trivial() { | |
| %0 = memref.alloc() : memref<10xf32> | |
| %c0 = arith.constant 0 : index | |
| %cst = arith.constant 0.000000e+00 : f32 | |
| affine.for %i1 = 0 to 10 { | |
| %1 = affine.apply affine_map<()[s0] -> (s0)>()[%c0] | |
| memref.store %cst, %0[%1] : memref<10xf32> | |
| %2 = memref.load %0[%c0] : memref<10xf32> | |
| %3 = affine.apply affine_map<()[] -> (0)>()[] | |
| memref.store %cst, %0[%3] : memref<10xf32> | |
| memref.store %2, %0[%c0] : memref<10xf32> | |
| } | |
| return | |
| } |