// RUN: mlir-query %s -c "m getUsersByPredicate(anyOf(hasOpName(\"memref.alloc\"),isConstantOp()),anyOf(hasOpName(\"affine.load\"), hasOpName(\"memref.dealloc\")),true)" | FileCheck %s

func.func @slice_depth1_loop_nest_with_offsets() {
  %0 = memref.alloc() : memref<100xf32>
  %cst = arith.constant 7.000000e+00 : f32
  affine.for %i0 = 0 to 16 {
    %a0 = affine.apply affine_map<(d0) -> (d0 + 2)>(%i0)
    affine.store %cst, %0[%a0] : memref<100xf32>
  }
  affine.for %i1 = 4 to 8 {
    %a1 = affine.apply affine_map<(d0) -> (d0 - 1)>(%i1)
    %1 = affine.load %0[%a1] : memref<100xf32>
  }
  return
}

// CHECK: Match #1:
// CHECK: {{.*}}.mlir:4:8: note: "root" binds here
// CHECK: %[[ALLOC:.*]] = memref.alloc() : memref<100xf32>

// CHECK: affine.store %cst, %0[%a0] : memref<100xf32>

// CHECK: Match #2:
// CHECK: {{.*}}.mlir:5:10: note: "root" binds here
// CHECK: %[[CST:.*]] = arith.constant 7.000000e+00 : f32

// CHECK: affine.store %[[CST]], %0[%a0] : memref<100xf32>