llvm/test/CodeGen/AMDGPU/returnaddress.ll

   1 ; RUN: llc -mtriple=amdgcn-amd-amdhsa -mcpu=gfx900 -verify-machineinstrs < %s | FileCheck --check-prefix=GCN %s
   2 ; RUN: llc -global-isel -mtriple=amdgcn-amd-amdhsa -mcpu=gfx900 -verify-machineinstrs < %s | FileCheck --check-prefix=GCN %s
   3
   4 ; Test with zero frame
   5 ; GCN-LABEL: {{^}}func1
   6 ; GCN: v_mov_b32_e32 v0, s30
   7 ; GCN: v_mov_b32_e32 v1, s31
   8 ; GCN: s_setpc_b64 s[30:31]
   9 define ptr @func1() nounwind {
  10 entry:
  11   %0 = tail call ptr @llvm.returnaddress(i32 0)
  12   ret ptr %0
  13 }
  14
  15 ; Test with non-zero frame
  16 ; GCN-LABEL: {{^}}func2
  17 ; GCN: v_mov_b32_e32 v0, 0
  18 ; GCN: v_mov_b32_e32 v1, 0
  19 ; GCN: s_setpc_b64 s[30:31]
  20 define ptr @func2() nounwind {
  21 entry:
  22   %0 = tail call ptr @llvm.returnaddress(i32 1)
  23   ret ptr %0
  24 }
  25
  26 ; Test with amdgpu_kernel
  27 ; GCN-LABEL: {{^}}func3
  28 ; GCN: v_mov_b32_e32 v0, 0
  29 ; GCN: v_mov_b32_e32 v1, {{v0|0}}
  30 define amdgpu_kernel void @func3(ptr %out) nounwind {
  31 entry:
  32   %tmp = tail call ptr @llvm.returnaddress(i32 0)
  33   store ptr %tmp, ptr %out, align 4
  34   ret void
  35 }
  36
  37 ; Test with use outside the entry-block
  38 ; GCN-LABEL: {{^}}func4
  39 ; GCN: v_mov_b32_e32 v0, 0
  40 ; GCN: v_mov_b32_e32 v1, {{v0|0}}
  41 define amdgpu_kernel void @func4(ptr %out, i32 %val) nounwind {
  42 entry:
  43   %cmp = icmp ne i32 %val, 0
  44   br i1 %cmp, label %store, label %exit
  45
  46 store:
  47   %tmp = tail call ptr @llvm.returnaddress(i32 1)
  48   store ptr %tmp, ptr %out, align 4
  49   ret void
  50
  51 exit:
  52   ret void
  53 }
  54
  55 ; Test ending in unreachable
  56 ; GCN-LABEL: {{^}}func5
  57 ; GCN: v_mov_b32_e32 v0, 0
  58 define void @func5() nounwind {
  59 entry:
  60   %tmp = tail call ptr @llvm.returnaddress(i32 2)
  61   store volatile i32 0, ptr addrspace(3) undef, align 4
  62   unreachable
  63 }
  64
  65 declare void @callee()
  66
  67 ; GCN-LABEL: {{^}}multi_use:
  68 ; GCN-DAG: v_mov_b32_e32 v[[LO:4[0-9]+]], s30
  69 ; GCN-DAG: v_mov_b32_e32 v[[HI:4[0-9]+]], s31
  70 ; GCN: global_store_dwordx2 v{{\[[0-9]+:[0-9]+\]}}, v[[[LO]]:[[HI]]]
  71 ; GCN: s_swappc_b64
  72 ; GCN: global_store_dwordx2 v{{\[[0-9]+:[0-9]+\]}}, v[[[LO]]:[[HI]]]
  73 define void @multi_use() nounwind {
  74 entry:
  75   %ret0 = tail call ptr @llvm.returnaddress(i32 0)
  76   store volatile ptr %ret0, ptr addrspace(1) undef
  77   call void @callee()
  78   %ret1 = tail call ptr @llvm.returnaddress(i32 0)
  79   store volatile ptr %ret1, ptr addrspace(1) undef
  80   ret void
  81 }
  82
  83 declare ptr @llvm.returnaddress(i32) nounwind readnone