8913b35f08
As a result, we have identical alignment calculation of byval
parameters for:
- LowerCall() - getting alignment of an argument (.param)
- emitFunctionParamList() - getting alignment of a
parameter (.param) in a function declaration
- getPrototype() - getting alignment of a parameter (.param) in a
function prototypes that is used for indirect calls
This change is required to avoid ptxas error: 'Alignment of argument
does not match formal parameter'. This error happens even in cases
where it logically shouldn't.
For instance:
.param .align 4 .b8 param0[4];
...
callprototype ()_ (.param .align 2 .b8 _[4]);
...
Here we allocate 'param0' with alignment of 4 and it should be fine to
pass it to a function that requires minimum alignment of 2.
At least ptxas v12.0 rejects this code.
Differential Revision: https://reviews.llvm.org/D140581
67 lines
1.8 KiB
LLVM
67 lines
1.8 KiB
LLVM
; RUN: llc < %s -march=nvptx -mcpu=sm_20 | FileCheck %s
|
|
; RUN: %if ptxas %{ llc < %s -march=nvptx -mcpu=sm_20 | %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
|
|
}
|
|
|
|
;;; Need at least 4-byte alignment in order to avoid miscompilation by
|
|
;;; ptxas for sm_50+
|
|
define ptx_device void @t4(ptr byval(i8) %x) {
|
|
; CHECK: .func t4
|
|
; CHECK: .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
|
|
; CHECK: .param .align 4 .b8 t5_param_0[1]
|
|
; CHECK: {
|
|
; CHECK: .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)
|
|
; CHECK: prototype_4 : .callprototype ()_ (.param .align 4 .b8 _[4]);
|
|
call void %fp(ptr byval(%struct.half2) null);
|
|
|
|
%fp3 = call ptr @getfp(i32 2)
|
|
; CHECK: prototype_6 : .callprototype ()_ (.param .align 4 .b8 _[1]);
|
|
call void %fp(ptr byval(i8) null);
|
|
ret void
|
|
}
|