llvm/test/CodeGen/AMDGPU/lower-buffer-fat-pointers-calls.ll

   1 ; NOTE: Assertions have been autogenerated by utils/update_test_checks.py UTC_ARGS: --version 2
   2 ; RUN: opt -S -mcpu=gfx900 -amdgpu-lower-buffer-fat-pointers < %s | FileCheck %s
   3 ; RUN: opt -S -mcpu=gfx900 -passes=amdgpu-lower-buffer-fat-pointers < %s | FileCheck %s
   4
   5 target datalayout = "e-p:64:64-p1:64:64-p2:32:32-p3:32:32-p4:64:64-p5:32:32-p6:32:32-p7:160:256:256:32-p8:128:128-i64:64-v16:16-v24:32-v32:32-v48:64-v96:128-v192:256-v256:256-v512:512-v1024:1024-v2048:2048-n32:64-S32-A5-G1-ni:7:8"
   6 target triple = "amdgcn--"
   7
   8 define ptr addrspace(7) @recur.inner.1(ptr addrspace(7) %x, i32 %v) {
   9 ; CHECK-LABEL: define { ptr addrspace(8), i32 } @recur.inner.1
  10 ; CHECK-SAME: ({ ptr addrspace(8), i32 } [[X:%.*]], i32 [[V:%.*]]) #[[ATTR0:[0-9]+]] {
  11 ; CHECK-NEXT:  bb:
  12 ; CHECK-NEXT:    [[X_RSRC:%.*]] = extractvalue { ptr addrspace(8), i32 } [[X]], 0
  13 ; CHECK-NEXT:    [[X_OFF:%.*]] = extractvalue { ptr addrspace(8), i32 } [[X]], 1
  14 ; CHECK-NEXT:    [[ISBASE:%.*]] = icmp sgt i32 [[V]], 0
  15 ; CHECK-NEXT:    br i1 [[ISBASE]], label [[RECUR:%.*]], label [[ELSE:%.*]]
  16 ; CHECK:       recur:
  17 ; CHECK-NEXT:    [[DEC:%.*]] = sub i32 [[V]], 1
  18 ; CHECK-NEXT:    [[INC:%.*]] = call { ptr addrspace(8), i32 } @recur.inner.2(i32 [[DEC]], { ptr addrspace(8), i32 } [[X]])
  19 ; CHECK-NEXT:    [[INC_RSRC:%.*]] = extractvalue { ptr addrspace(8), i32 } [[INC]], 0
  20 ; CHECK-NEXT:    [[INC_OFF:%.*]] = extractvalue { ptr addrspace(8), i32 } [[INC]], 1
  21 ; CHECK-NEXT:    br label [[END:%.*]]
  22 ; CHECK:       else:
  23 ; CHECK-NEXT:    br label [[END]]
  24 ; CHECK:       end:
  25 ; CHECK-NEXT:    [[RET_RSRC:%.*]] = phi ptr addrspace(8) [ [[INC_RSRC]], [[RECUR]] ], [ [[X_RSRC]], [[ELSE]] ]
  26 ; CHECK-NEXT:    [[RET_OFF:%.*]] = phi i32 [ [[INC_OFF]], [[RECUR]] ], [ [[X_OFF]], [[ELSE]] ]
  27 ; CHECK-NEXT:    [[TMP0:%.*]] = insertvalue { ptr addrspace(8), i32 } poison, ptr addrspace(8) [[RET_RSRC]], 0
  28 ; CHECK-NEXT:    [[RET:%.*]] = insertvalue { ptr addrspace(8), i32 } [[TMP0]], i32 [[RET_OFF]], 1
  29 ; CHECK-NEXT:    ret { ptr addrspace(8), i32 } [[RET]]
  30 ;
  31 bb:
  32   %isBase = icmp sgt i32 %v, 0
  33   br i1 %isBase, label %recur, label %else
  34 recur:
  35   %dec = sub i32 %v, 1
  36   %inc = call ptr addrspace(7) @recur.inner.2(i32 %dec, ptr addrspace(7) %x)
  37   br label %end
  38 else:
  39   br label %end
  40 end:
  41   %ret = phi ptr addrspace(7) [%inc, %recur], [%x, %else]
  42   ret ptr addrspace(7) %ret
  43 }
  44
  45 define ptr addrspace(7) @recur.inner.2(i32 %v, ptr addrspace(7) %x) {
  46 ; CHECK-LABEL: define { ptr addrspace(8), i32 } @recur.inner.2
  47 ; CHECK-SAME: (i32 [[V:%.*]], { ptr addrspace(8), i32 } [[X:%.*]]) #[[ATTR0]] {
  48 ; CHECK-NEXT:    [[X_RSRC:%.*]] = extractvalue { ptr addrspace(8), i32 } [[X]], 0
  49 ; CHECK-NEXT:    [[X_OFF:%.*]] = extractvalue { ptr addrspace(8), i32 } [[X]], 1
  50 ; CHECK-NEXT:    [[INC:%.*]] = add i32 [[X_OFF]], 4
  51 ; CHECK-NEXT:    [[TMP1:%.*]] = insertvalue { ptr addrspace(8), i32 } poison, ptr addrspace(8) [[X_RSRC]], 0
  52 ; CHECK-NEXT:    [[TMP2:%.*]] = insertvalue { ptr addrspace(8), i32 } [[TMP1]], i32 [[INC]], 1
  53 ; CHECK-NEXT:    [[RET:%.*]] = call { ptr addrspace(8), i32 } @recur.inner.1({ ptr addrspace(8), i32 } [[TMP2]], i32 [[V]])
  54 ; CHECK-NEXT:    ret { ptr addrspace(8), i32 } [[RET]]
  55 ;
  56   %inc = getelementptr i32, ptr addrspace(7) %x, i32 1
  57   %ret = call ptr addrspace(7) @recur.inner.1(ptr addrspace(7) %inc, i32 %v)
  58   ret ptr addrspace(7) %ret
  59 }
  60
  61 define void @recur.outer(ptr addrspace(7) %x, ptr %arg) {
  62 ; CHECK-LABEL: define void @recur.outer
  63 ; CHECK-SAME: ({ ptr addrspace(8), i32 } [[X:%.*]], ptr [[ARG:%.*]]) #[[ATTR0]] {
  64 ; CHECK-NEXT:    [[BOUND:%.*]] = load i32, ptr [[ARG]], align 4
  65 ; CHECK-NEXT:    [[RET:%.*]] = call { ptr addrspace(8), i32 } @recur.inner.1({ ptr addrspace(8), i32 } [[X]], i32 [[BOUND]])
  66 ; CHECK-NEXT:    [[RET_RSRC:%.*]] = extractvalue { ptr addrspace(8), i32 } [[RET]], 0
  67 ; CHECK-NEXT:    [[RET_OFF:%.*]] = extractvalue { ptr addrspace(8), i32 } [[RET]], 1
  68 ; CHECK-NEXT:    [[RET_INT_RSRC:%.*]] = ptrtoint ptr addrspace(8) [[RET_RSRC]] to i160
  69 ; CHECK-NEXT:    [[TMP1:%.*]] = shl nuw i160 [[RET_INT_RSRC]], 32
  70 ; CHECK-NEXT:    [[RET_INT_OFF:%.*]] = zext i32 [[RET_OFF]] to i160
  71 ; CHECK-NEXT:    [[RET_INT:%.*]] = or i160 [[TMP1]], [[RET_INT_OFF]]
  72 ; CHECK-NEXT:    store i160 [[RET_INT]], ptr [[ARG]], align 32
  73 ; CHECK-NEXT:    ret void
  74 ;
  75   %bound = load i32, ptr %arg
  76   %ret = call ptr addrspace(7) @recur.inner.1(ptr addrspace(7) %x, i32 %bound)
  77   store ptr addrspace(7) %ret, ptr %arg
  78   ret void
  79 }
  80
  81 declare ptr addrspace(7) @extern(ptr addrspace(7) %arg)
  82 define void @caller(ptr addrspace(7) noundef nonnull %arg) {
  83 ; CHECK-LABEL: define void @caller
  84 ; CHECK-SAME: ({ ptr addrspace(8), i32 } noundef [[ARG:%.*]]) #[[ATTR0]] {
  85 ; CHECK-NEXT:    [[ARG_RSRC:%.*]] = extractvalue { ptr addrspace(8), i32 } [[ARG]], 0
  86 ; CHECK-NEXT:    [[ARG_OFF:%.*]] = extractvalue { ptr addrspace(8), i32 } [[ARG]], 1
  87 ; CHECK-NEXT:    [[V:%.*]] = call { ptr addrspace(8), i32 } @extern({ ptr addrspace(8), i32 } [[ARG]])
  88 ; CHECK-NEXT:    [[V_RSRC:%.*]] = extractvalue { ptr addrspace(8), i32 } [[V]], 0
  89 ; CHECK-NEXT:    [[V_OFF:%.*]] = extractvalue { ptr addrspace(8), i32 } [[V]], 1
  90 ; CHECK-NEXT:    [[V_INT_RSRC:%.*]] = ptrtoint ptr addrspace(8) [[V_RSRC]] to i160
  91 ; CHECK-NEXT:    [[TMP1:%.*]] = shl nuw i160 [[V_INT_RSRC]], 32
  92 ; CHECK-NEXT:    [[V_INT_OFF:%.*]] = zext i32 [[V_OFF]] to i160
  93 ; CHECK-NEXT:    [[V_INT:%.*]] = or i160 [[TMP1]], [[V_INT_OFF]]
  94 ; CHECK-NEXT:    call void @llvm.amdgcn.raw.ptr.buffer.store.i160(i160 [[V_INT]], ptr addrspace(8) align 32 [[ARG_RSRC]], i32 [[ARG_OFF]], i32 0, i32 0)
  95 ; CHECK-NEXT:    ret void
  96 ;
  97   %v = call ptr addrspace(7) @extern(ptr addrspace(7) %arg)
  98   store ptr addrspace(7) %v, ptr addrspace(7) %arg
  99   ret void
 100 }
 101
 102 define internal noalias noundef nonnull ptr addrspace(7) @foo(ptr addrspace(7) noalias noundef nonnull %arg) {
 103 ; CHECK-LABEL: define internal noundef { ptr addrspace(8), i32 } @foo
 104 ; CHECK-SAME: ({ ptr addrspace(8), i32 } noundef [[ARG:%.*]]) #[[ATTR0]] {
 105 ; CHECK-NEXT:    [[ARG_RSRC:%.*]] = extractvalue { ptr addrspace(8), i32 } [[ARG]], 0
 106 ; CHECK-NEXT:    [[ARG_OFF:%.*]] = extractvalue { ptr addrspace(8), i32 } [[ARG]], 1
 107 ; CHECK-NEXT:    [[RET:%.*]] = add nuw i32 [[ARG_OFF]], 4
 108 ; CHECK-NEXT:    [[TMP1:%.*]] = insertvalue { ptr addrspace(8), i32 } poison, ptr addrspace(8) [[ARG_RSRC]], 0
 109 ; CHECK-NEXT:    [[TMP2:%.*]] = insertvalue { ptr addrspace(8), i32 } [[TMP1]], i32 [[RET]], 1
 110 ; CHECK-NEXT:    ret { ptr addrspace(8), i32 } [[TMP2]]
 111 ;
 112   %ret = getelementptr inbounds i32, ptr addrspace(7) %arg, i32 1
 113   ret ptr addrspace(7) %ret
 114 }