1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
|
; RUN: llc < %s -march=nvptx64 -mcpu=sm_20 | FileCheck %s --check-prefixes=CHECK,NOALIGN4
; RUN: llc < %s -march=nvptx64 -mcpu=sm_20 -nvptx-force-min-byval-param-align | FileCheck %s --check-prefixes=CHECK,ALIGN4
; RUN: %if ptxas %{ llc < %s -march=nvptx64 -mcpu=sm_20 | %ptxas-verify %}
; RUN: %if ptxas %{ llc < %s -march=nvptx64 -mcpu=sm_20 -nvptx-force-min-byval-param-align | %ptxas-verify %}
;;; Need 4-byte alignment on ptr passed byval
define ptx_device void @t1(ptr byval(float) %x) {
; CHECK: .func t1
; CHECK: .param .align 4 .b8 t1_param_0[4]
ret void
}
;;; Need 8-byte alignment on ptr passed byval
define ptx_device void @t2(ptr byval(double) %x) {
; CHECK: .func t2
; CHECK: .param .align 8 .b8 t2_param_0[8]
ret void
}
;;; Need 4-byte alignment on float2* passed byval
%struct.float2 = type { float, float }
define ptx_device void @t3(ptr byval(%struct.float2) %x) {
; CHECK: .func t3
; CHECK: .param .align 4 .b8 t3_param_0[8]
ret void
}
define ptx_device void @t4(ptr byval(i8) %x) {
; CHECK: .func t4
; NOALIGN4: .param .align 1 .b8 t4_param_0[1]
; ALIGN4: .param .align 4 .b8 t4_param_0[1]
ret void
}
;;; Make sure we adjust alignment at the call site as well.
define ptx_device void @t5(ptr align 2 byval(i8) %x) {
; CHECK: .func t5
; NOALIGN4: .param .align 2 .b8 t5_param_0[1]
; ALIGN4: .param .align 4 .b8 t5_param_0[1]
; CHECK: {
; NOALIGN4: .param .align 1 .b8 param0[1];
; ALIGN4: .param .align 4 .b8 param0[1];
; CHECK: call.uni
call void @t4(ptr byval(i8) %x)
ret void
}
;;; Make sure we adjust alignment for a function prototype
;;; in case of an inderect call.
declare ptr @getfp(i32 %n)
%struct.half2 = type { half, half }
define ptx_device void @t6() {
; CHECK: .func t6
%fp = call ptr @getfp(i32 0)
; CHECK: prototype_2 : .callprototype ()_ (.param .align 8 .b8 _[8]);
call void %fp(ptr byval(double) null);
%fp2 = call ptr @getfp(i32 1)
; NOALIGN4: prototype_4 : .callprototype ()_ (.param .align 2 .b8 _[4]);
; ALIGN4: prototype_4 : .callprototype ()_ (.param .align 4 .b8 _[4]);
call void %fp(ptr byval(%struct.half2) null);
%fp3 = call ptr @getfp(i32 2)
; NOALIGN4: prototype_6 : .callprototype ()_ (.param .align 1 .b8 _[1]);
; ALIGN4: prototype_6 : .callprototype ()_ (.param .align 4 .b8 _[1]);
call void %fp(ptr byval(i8) null);
ret void
}
; CHECK-LABEL: .func check_ptr_align1(
; CHECK: ld.param.u64 %rd1, [check_ptr_align1_param_0];
; CHECK-NOT: ld.param.u8
; CHECK: mov.b32 %r1, 0;
; CHECK: st.u8 [%rd1+3], %r1;
; CHECK: st.u8 [%rd1+2], %r1;
; CHECK: st.u8 [%rd1+1], %r1;
; CHECK: mov.b32 %r2, 1;
; CHECK: st.u8 [%rd1], %r2;
; CHECK: ret;
define void @check_ptr_align1(ptr align 1 %_arg_ptr) {
entry:
store i32 1, ptr %_arg_ptr, align 1
ret void
}
; CHECK-LABEL: .func check_ptr_align2(
; CHECK: ld.param.u64 %rd1, [check_ptr_align2_param_0];
; CHECK-NOT: ld.param.u16
; CHECK: mov.b32 %r1, 0;
; CHECK: st.u16 [%rd1+2], %r1;
; CHECK: mov.b32 %r2, 2;
; CHECK: st.u16 [%rd1], %r2;
; CHECK: ret;
define void @check_ptr_align2(ptr align 2 %_arg_ptr) {
entry:
store i32 2, ptr %_arg_ptr, align 2
ret void
}
; CHECK-LABEL: .func check_ptr_align4(
; CHECK: ld.param.u64 %rd1, [check_ptr_align4_param_0];
; CHECK-NOT: ld.param.u32
; CHECK: mov.b32 %r1, 4;
; CHECK: st.u32 [%rd1], %r1;
; CHECK: ret;
define void @check_ptr_align4(ptr align 4 %_arg_ptr) {
entry:
store i32 4, ptr %_arg_ptr, align 4
ret void
}
; CHECK-LABEL: .func check_ptr_align8(
; CHECK: ld.param.u64 %rd1, [check_ptr_align8_param_0];
; CHECK: mov.b32 %r1, 8;
; CHECK: st.u32 [%rd1], %r1;
; CHECK: ret;
define void @check_ptr_align8(ptr align 8 %_arg_ptr) {
entry:
store i32 8, ptr %_arg_ptr, align 8
ret void
}
|
