llvm/test/CodeGen/NVPTX/param-align.ll

   1 ; RUN: llc < %s -march=nvptx64 -mcpu=sm_20 | FileCheck %s --check-prefixes=CHECK,NOALIGN4
   2 ; RUN: llc < %s -march=nvptx64 -mcpu=sm_20 -nvptx-force-min-byval-param-align | FileCheck %s --check-prefixes=CHECK,ALIGN4
   3 ; RUN: %if ptxas %{ llc < %s -march=nvptx64 -mcpu=sm_20 | %ptxas-verify %}
   4 ; RUN: %if ptxas %{ llc < %s -march=nvptx64 -mcpu=sm_20 -nvptx-force-min-byval-param-align | %ptxas-verify %}
   5
   6 ;;; Need 4-byte alignment on ptr passed byval
   7 define ptx_device void @t1(ptr byval(float) %x) {
   8 ; CHECK: .func t1
   9 ; CHECK: .param .align 4 .b8 t1_param_0[4]
  10   ret void
  11 }
  12
  13
  14 ;;; Need 8-byte alignment on ptr passed byval
  15 define ptx_device void @t2(ptr byval(double) %x) {
  16 ; CHECK: .func t2
  17 ; CHECK: .param .align 8 .b8 t2_param_0[8]
  18   ret void
  19 }
  20
  21
  22 ;;; Need 4-byte alignment on float2* passed byval
  23 %struct.float2 = type { float, float }
  24 define ptx_device void @t3(ptr byval(%struct.float2) %x) {
  25 ; CHECK: .func t3
  26 ; CHECK: .param .align 4 .b8 t3_param_0[8]
  27   ret void
  28 }
  29
  30 define ptx_device void @t4(ptr byval(i8) %x) {
  31 ; CHECK: .func t4
  32 ; NOALIGN4: .param .align 1 .b8 t4_param_0[1]
  33 ; ALIGN4: .param .align 4 .b8 t4_param_0[1]
  34   ret void
  35 }
  36
  37 ;;; Make sure we adjust alignment at the call site as well.
  38 define ptx_device void @t5(ptr align 2 byval(i8) %x) {
  39 ; CHECK: .func t5
  40 ; NOALIGN4: .param .align 2 .b8 t5_param_0[1]
  41 ; ALIGN4: .param .align 4 .b8 t5_param_0[1]
  42 ; CHECK: {
  43 ; NOALIGN4: .param .align 1 .b8 param0[1];
  44 ; ALIGN4:   .param .align 4 .b8 param0[1];
  45 ; CHECK: call.uni
  46   call void @t4(ptr byval(i8) %x)
  47   ret void
  48 }
  49
  50 ;;; Make sure we adjust alignment for a function prototype
  51 ;;; in case of an inderect call.
  52
  53 declare ptr @getfp(i32 %n)
  54 %struct.half2 = type { half, half }
  55 define ptx_device void @t6() {
  56 ; CHECK: .func t6
  57   %fp = call ptr @getfp(i32 0)
  58 ; CHECK: prototype_2 : .callprototype ()_ (.param .align 8 .b8 _[8]);
  59   call void %fp(ptr byval(double) null);
  60
  61   %fp2 = call ptr @getfp(i32 1)
  62 ; NOALIGN4: prototype_4 : .callprototype ()_ (.param .align 2 .b8 _[4]);
  63 ; ALIGN4: prototype_4 : .callprototype ()_ (.param .align 4 .b8 _[4]);
  64   call void %fp(ptr byval(%struct.half2) null);
  65
  66   %fp3 = call ptr @getfp(i32 2)
  67 ; NOALIGN4: prototype_6 : .callprototype ()_ (.param .align 1 .b8 _[1]);
  68 ; ALIGN4: prototype_6 : .callprototype ()_ (.param .align 4 .b8 _[1]);
  69   call void %fp(ptr byval(i8) null);
  70   ret void
  71 }