llvm/flang/test/Transforms/stack-reclaime.fir

// RUN: fir-opt --split-input-file --stack-reclaim %s | FileCheck %s

func.func @alloca_in_loop(%lb : index, %ub : index, %step : index, %b : i1, %addr : !fir.ref<index>) {
  fir.do_loop %iv = %lb to %ub step %step unordered {
    %0 = fir.alloca !fir.box<!fir.heap<!fir.char<1,?>>>
  }
  return
}

// CHECK-LABEL: func.func @alloca_in_loop
// CHECK: fir.do_loop
// CHECK: %[[STACKPTR:.*]] = llvm.intr.stacksave : !llvm.ptr
// CHECK: %{{.*}} = fir.alloca !fir.box<!fir.heap<!fir.char<1,?>>>
// CHECK: llvm.intr.stackrestore %0 : !llvm.ptr

// -----

module attributes {dlti.dl_spec = #dlti.dl_spec<#dlti.dl_entry<"dlti.alloca_memory_space", 5 : ui64>>} {
  func.func @stack_restore_save_alloca_address(%lb : index, %ub : index, %step : index, %b : i1, %addr : !fir.ref<index>) {
    fir.do_loop %iv = %lb to %ub step %step unordered {
      %0 = fir.alloca !fir.box<!fir.heap<!fir.char<1,?>>>
    }
    return
  }
}

// CHECK-LABEL: func.func @stack_restore_save_alloca_address
// CHECK: fir.do_loop
// CHECK: %[[STACKPTR:.*]] = llvm.intr.stacksave : !llvm.ptr<5>
// CHECK: %{{.*}} = fir.alloca !fir.box<!fir.heap<!fir.char<1,?>>>
// CHECK: llvm.intr.stackrestore %0 : !llvm.ptr<5>