llvm/test/CodeGen/AMDGPU/replace-lds-by-ptr-indirect-call-signature-match.ll

   1 ; RUN: opt -S -mtriple=amdgcn--  -amdgpu-replace-lds-use-with-pointer -amdgpu-enable-lds-replace-with-pointer=true < %s | FileCheck %s
   2
   3 ; DESCRIPTION:
   4 ;
   5 ; There are three lds globals defined here, and these three lds are used respectively within
   6 ; three non-kernel functions. There is one kernel which *indirectly* calls one of the non-kernel
   7 ; functions. But since all the three non-kernel functions have same signature, all three
   8 ; non-kernel functions are resolved as potential callees for indirect call-site. Hence we land-up
   9 ; pointer replacement for three lds globals.
  10 ;
  11
  12 ; Original LDS should exist.
  13 ; CHECK: @lds_used_within_function_1 = internal addrspace(3) global [4 x i32] undef, align 4
  14 ; CHECK: @lds_used_within_function_2 = internal addrspace(3) global [4 x i32] undef, align 4
  15 ; CHECK: @lds_used_within_function_3 = internal addrspace(3) global [4 x i32] undef, align 4
  16 @lds_used_within_function_1 = internal addrspace(3) global [4 x i32] undef, align 4
  17 @lds_used_within_function_2 = internal addrspace(3) global [4 x i32] undef, align 4
  18 @lds_used_within_function_3 = internal addrspace(3) global [4 x i32] undef, align 4
  19
  20 ; Function pointers should exist.
  21 ; CHECK: @ptr_to_func1 = internal local_unnamed_addr externally_initialized global void (i16)* @function_1, align 8
  22 ; CHECK: @ptr_to_func2 = internal local_unnamed_addr externally_initialized global void (i16)* @function_2, align 8
  23 ; CHECK: @ptr_to_func3 = internal local_unnamed_addr externally_initialized global void (i16)* @function_3, align 8
  24 @ptr_to_func1 = internal local_unnamed_addr externally_initialized global void (i16)* @function_1, align 8
  25 @ptr_to_func2 = internal local_unnamed_addr externally_initialized global void (i16)* @function_2, align 8
  26 @ptr_to_func3 = internal local_unnamed_addr externally_initialized global void (i16)* @function_3, align 8
  27
  28 ; Pointers should be created.
  29 ; CHECK: @lds_used_within_function_1.ptr = internal unnamed_addr addrspace(3) global i16 undef, align 2
  30 ; CHECK: @lds_used_within_function_2.ptr = internal unnamed_addr addrspace(3) global i16 undef, align 2
  31 ; CHECK: @lds_used_within_function_3.ptr = internal unnamed_addr addrspace(3) global i16 undef, align 2
  32
  33 ; Pointer replacement code should be added.
  34 define internal void @function_3(i16 %i) {
  35 ; CHECK-LABEL: entry:
  36 ; CHECK:   %0 = load i16, i16 addrspace(3)* @lds_used_within_function_3.ptr, align 2
  37 ; CHECK:   %1 = getelementptr i8, i8 addrspace(3)* null, i16 %0
  38 ; CHECK:   %2 = bitcast i8 addrspace(3)* %1 to [4 x i32] addrspace(3)*
  39 ; CHECK:   %gep = getelementptr inbounds [4 x i32], [4 x i32] addrspace(3)* %2, i32 0, i32 0
  40 ; CHECK:   ret void
  41 entry:
  42   %gep = getelementptr inbounds [4 x i32], [4 x i32] addrspace(3)* @lds_used_within_function_3, i32 0, i32 0
  43   ret void
  44 }
  45
  46 ; Pointer replacement code should be added.
  47 define internal void @function_2(i16 %i) {
  48 ; CHECK-LABEL: entry:
  49 ; CHECK:   %0 = load i16, i16 addrspace(3)* @lds_used_within_function_2.ptr, align 2
  50 ; CHECK:   %1 = getelementptr i8, i8 addrspace(3)* null, i16 %0
  51 ; CHECK:   %2 = bitcast i8 addrspace(3)* %1 to [4 x i32] addrspace(3)*
  52 ; CHECK:   %gep = getelementptr inbounds [4 x i32], [4 x i32] addrspace(3)* %2, i32 0, i32 0
  53 ; CHECK:   ret void
  54 entry:
  55   %gep = getelementptr inbounds [4 x i32], [4 x i32] addrspace(3)* @lds_used_within_function_2, i32 0, i32 0
  56   ret void
  57 }
  58
  59 ; Pointer replacement code should be added.
  60 define internal void @function_1(i16 %i) {
  61 ; CHECK-LABEL: entry:
  62 ; CHECK:   %0 = load i16, i16 addrspace(3)* @lds_used_within_function_1.ptr, align 2
  63 ; CHECK:   %1 = getelementptr i8, i8 addrspace(3)* null, i16 %0
  64 ; CHECK:   %2 = bitcast i8 addrspace(3)* %1 to [4 x i32] addrspace(3)*
  65 ; CHECK:   %gep = getelementptr inbounds [4 x i32], [4 x i32] addrspace(3)* %2, i32 0, i32 0
  66 ; CHECK:   ret void
  67 entry:
  68   %gep = getelementptr inbounds [4 x i32], [4 x i32] addrspace(3)* @lds_used_within_function_1, i32 0, i32 0
  69   ret void
  70 }
  71
  72 ; Pointer initialization code shoud be added
  73 define protected amdgpu_kernel void @kernel_indirectly_calls_function_1() {
  74 ; CHECK-LABEL: entry:
  75 ; CHECK:   %0 = call i32 @llvm.amdgcn.mbcnt.lo(i32 -1, i32 0)
  76 ; CHECK:   %1 = icmp eq i32 %0, 0
  77 ; CHECK:   br i1 %1, label %2, label %3
  78 ;
  79 ; CHECK-LABEL: 2:
  80 ; CHECK:   store i16 ptrtoint ([4 x i32] addrspace(3)* @lds_used_within_function_3 to i16), i16 addrspace(3)* @lds_used_within_function_3.ptr, align 2
  81 ; CHECK:   store i16 ptrtoint ([4 x i32] addrspace(3)* @lds_used_within_function_2 to i16), i16 addrspace(3)* @lds_used_within_function_2.ptr, align 2
  82 ; CHECK:   store i16 ptrtoint ([4 x i32] addrspace(3)* @lds_used_within_function_1 to i16), i16 addrspace(3)* @lds_used_within_function_1.ptr, align 2
  83 ; CHECK:   br label %3
  84 ;
  85 ; CHECK-LABEL: 3:
  86 ; CHECK:   call void @llvm.amdgcn.wave.barrier()
  87 ; CHECK:   %fptr1 = load void (i16)*, void (i16)** @ptr_to_func1, align 8
  88 ; CHECK:   call void %fptr1(i16 6)
  89 ; CHECK:   ret void
  90 entry:
  91   %fptr1 = load void (i16)*, void (i16)** @ptr_to_func1, align 8
  92   call void %fptr1(i16 6)
  93   ret void
  94 }