llvm/test/CodeGen/AArch64/GlobalISel/irtranslator-localescape.ll

   1 ; NOTE: Assertions have been autogenerated by utils/update_mir_test_checks.py
   2 ; RUN: llc -global-isel -mtriple=arm64-windows -stop-after=irtranslator -o - %s | FileCheck %s
   3
   4 define void @local_escape() {
   5   ; CHECK-LABEL: name: local_escape
   6   ; CHECK: bb.1 (%ir-block.0):
   7   ; CHECK:   LOCAL_ESCAPE <mcsymbol .Llocal_escape$frame_escape_1>, %stack.1.b
   8   ; CHECK:   LOCAL_ESCAPE <mcsymbol .Llocal_escape$frame_escape_0>, %stack.0.a
   9   ; CHECK:   [[C:%[0-9]+]]:_(s32) = G_CONSTANT i32 42
  10   ; CHECK:   [[C1:%[0-9]+]]:_(s32) = G_CONSTANT i32 13
  11   ; CHECK:   [[FRAME_INDEX:%[0-9]+]]:_(p0) = G_FRAME_INDEX %stack.0.a
  12   ; CHECK:   [[FRAME_INDEX1:%[0-9]+]]:_(p0) = G_FRAME_INDEX %stack.1.b
  13   ; CHECK:   G_STORE [[C]](s32), [[FRAME_INDEX]](p0) :: (store (s32) into %ir.a)
  14   ; CHECK:   G_STORE [[C1]](s32), [[FRAME_INDEX1]](p0) :: (store (s32) into %ir.b)
  15   ; CHECK:   RET_ReallyLR
  16   %a = alloca i32
  17   %b = alloca i32, i32 2
  18   call void (...) @llvm.localescape(i32* %a, i32* %b)
  19   store i32 42, i32* %a
  20   store i32 13, i32* %b
  21   ret void
  22 }
  23
  24 ; Try some instructions before the localescape, and use a null
  25 define void @local_escape_insert_point() {
  26   ; CHECK-LABEL: name: local_escape_insert_point
  27   ; CHECK: bb.1 (%ir-block.0):
  28   ; CHECK:   LOCAL_ESCAPE <mcsymbol .Llocal_escape_insert_point$frame_escape_2>, %stack.1.b
  29   ; CHECK:   LOCAL_ESCAPE <mcsymbol .Llocal_escape_insert_point$frame_escape_0>, %stack.0.a
  30   ; CHECK:   [[C:%[0-9]+]]:_(s32) = G_CONSTANT i32 42
  31   ; CHECK:   [[C1:%[0-9]+]]:_(s32) = G_CONSTANT i32 13
  32   ; CHECK:   [[FRAME_INDEX:%[0-9]+]]:_(p0) = G_FRAME_INDEX %stack.0.a
  33   ; CHECK:   [[FRAME_INDEX1:%[0-9]+]]:_(p0) = G_FRAME_INDEX %stack.1.b
  34   ; CHECK:   G_STORE [[C]](s32), [[FRAME_INDEX]](p0) :: (store (s32) into %ir.a)
  35   ; CHECK:   G_STORE [[C1]](s32), [[FRAME_INDEX1]](p0) :: (store (s32) into %ir.b)
  36   ; CHECK:   RET_ReallyLR
  37   %a = alloca i32
  38   %b = alloca i32, i32 2
  39   store i32 42, i32* %a
  40   store i32 13, i32* %b
  41   call void (...) @llvm.localescape(i32* %a, i32* null, i32* %b)
  42   ret void
  43 }
  44
  45 declare void @foo([128 x i32]*)
  46
  47 ; Check a cast of an alloca
  48 define void @local_escape_strip_ptr_cast() {
  49   ; CHECK-LABEL: name: local_escape_strip_ptr_cast
  50   ; CHECK: bb.1 (%ir-block.0):
  51   ; CHECK:   LOCAL_ESCAPE <mcsymbol .Llocal_escape_strip_ptr_cast$frame_escape_0>, %stack.0.a
  52   ; CHECK:   [[C:%[0-9]+]]:_(s32) = G_CONSTANT i32 42
  53   ; CHECK:   [[FRAME_INDEX:%[0-9]+]]:_(p0) = G_FRAME_INDEX %stack.0.a
  54   ; CHECK:   G_STORE [[C]](s32), [[FRAME_INDEX]](p0) :: (store (s32) into %ir.cast)
  55   ; CHECK:   ADJCALLSTACKDOWN 0, 0, implicit-def $sp, implicit $sp
  56   ; CHECK:   $x0 = COPY [[FRAME_INDEX]](p0)
  57   ; CHECK:   BL @foo, csr_aarch64_aapcs, implicit-def $lr, implicit $sp, implicit $x0
  58   ; CHECK:   ADJCALLSTACKUP 0, 0, implicit-def $sp, implicit $sp
  59   ; CHECK:   RET_ReallyLR
  60   %a = alloca [128 x i32]
  61   %cast = bitcast [128 x i32]* %a to i32*
  62   store i32 42, i32* %cast
  63   call void (...) @llvm.localescape(i32* %cast, i32* null)
  64   call void @foo([128 x i32]* %a)
  65   ret void
  66 }
  67
  68 declare void @llvm.localescape(...) #0
  69
  70 attributes #0 = { nounwind }