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 }
  72
  73 ; CHECK-LABEL: .func check_ptr_align1(
  74 ; CHECK:        ld.param.u64    %rd1, [check_ptr_align1_param_0];
  75 ; CHECK-NOT:    ld.param.u8
  76 ; CHECK:        mov.b32         %r1, 0;
  77 ; CHECK:        st.u8   [%rd1+3], %r1;
  78 ; CHECK:        st.u8   [%rd1+2], %r1;
  79 ; CHECK:        st.u8   [%rd1+1], %r1;
  80 ; CHECK:        mov.b32         %r2, 1;
  81 ; CHECK:        st.u8   [%rd1], %r2;
  82 ; CHECK:        ret;
  83 define void @check_ptr_align1(ptr align 1 %_arg_ptr) {
  84 entry:
  85   store i32 1, ptr %_arg_ptr, align 1
  86   ret void
  87 }
  88
  89 ; CHECK-LABEL: .func check_ptr_align2(
  90 ; CHECK:        ld.param.u64    %rd1, [check_ptr_align2_param_0];
  91 ; CHECK-NOT:    ld.param.u16
  92 ; CHECK:        mov.b32         %r1, 0;
  93 ; CHECK:        st.u16  [%rd1+2], %r1;
  94 ; CHECK:        mov.b32         %r2, 2;
  95 ; CHECK:        st.u16  [%rd1], %r2;
  96 ; CHECK:        ret;
  97 define void @check_ptr_align2(ptr align 2 %_arg_ptr) {
  98 entry:
  99   store i32 2, ptr %_arg_ptr, align 2
 100   ret void
 101 }
 102
 103 ; CHECK-LABEL: .func check_ptr_align4(
 104 ; CHECK:        ld.param.u64    %rd1, [check_ptr_align4_param_0];
 105 ; CHECK-NOT:    ld.param.u32
 106 ; CHECK:        mov.b32         %r1, 4;
 107 ; CHECK:        st.u32  [%rd1], %r1;
 108 ; CHECK:        ret;
 109 define void @check_ptr_align4(ptr align 4 %_arg_ptr) {
 110 entry:
 111   store i32 4, ptr %_arg_ptr, align 4
 112   ret void
 113 }
 114
 115 ; CHECK-LABEL: .func check_ptr_align8(
 116 ; CHECK:        ld.param.u64    %rd1, [check_ptr_align8_param_0];
 117 ; CHECK:        mov.b32         %r1, 8;
 118 ; CHECK:        st.u32  [%rd1], %r1;
 119 ; CHECK:        ret;
 120 define void @check_ptr_align8(ptr align 8 %_arg_ptr) {
 121 entry:
 122   store i32 8, ptr %_arg_ptr, align 8
 123   ret void
 124 }