69e73d076b
When --nvptx-short-ptr is set, local pointers are stored as 32-bit on
nvptx64 target.
Before this patch, arguments for a function declaration were always
emitted as b64 regardless of their address space, but they were set as
b32 for the corresponding call instruction:
.extern .func test
(
.param .b64 test_param_0
)
[...]
.param .b32 param0;
st.param.b32 [param0+0], %r1;
call.uni test, (param0);
This is not supported:
ptxas: Type of argument does not match formal parameter
'test_param_0'
Now short pointers in a function declaration are emitted as b32 if
--nvptx-short-ptr is set.
Differential Revision: https://reviews.llvm.org/D135674
45 lines
1.8 KiB
LLVM
45 lines
1.8 KiB
LLVM
; RUN: llc < %s -march=nvptx64 -mcpu=sm_20 | FileCheck %s --check-prefix CHECK-DEFAULT
|
|
; RUN: llc < %s -march=nvptx -mcpu=sm_20 | FileCheck %s --check-prefix CHECK-DEFAULT-32
|
|
; RUN: llc < %s -march=nvptx64 -mcpu=sm_20 -nvptx-short-ptr | FileCheck %s --check-prefixes CHECK-SHORT-SHARED,CHECK-SHORT-CONST,CHECK-SHORT-LOCAL
|
|
|
|
; RUN: %if ptxas %{ llc < %s -march=nvptx -mcpu=sm_20 | %ptxas-verify %}
|
|
; RUN: %if ptxas %{ llc < %s -march=nvptx64 -mcpu=sm_20 | %ptxas-verify %}
|
|
; RUN: %if ptxas %{ llc < %s -march=nvptx64 -mcpu=sm_20 -nvptx-short-ptr | %ptxas-verify %}
|
|
|
|
; CHECK-DEFAULT: .visible .shared .align 8 .u64 s
|
|
; CHECK-DEFAULT-32: .visible .shared .align 8 .u32 s
|
|
; CHECK-SHORT-SHARED: .visible .shared .align 8 .u32 s
|
|
@s = local_unnamed_addr addrspace(3) global i32 addrspace(3)* null, align 8
|
|
|
|
; CHECK-DEFAULT: .visible .const .align 8 .u64 c
|
|
; CHECK-DEFAULT-32: .visible .const .align 8 .u32 c
|
|
; CHECK-SHORT-CONST: .visible .const .align 8 .u32 c
|
|
@c = local_unnamed_addr addrspace(4) global i32 addrspace(4)* null, align 8
|
|
|
|
declare void @use(i8 %arg);
|
|
|
|
; CHECK-DEFAULT: .param .b64 test1_param_0
|
|
; CHECK-DEFAULT-32: .param .b32 test1_param_0
|
|
; CHECK-SHORT-LOCAL: .param .b32 test1_param_0
|
|
define void @test1(i8 addrspace(5)* %local) {
|
|
; CHECK-DEFAULT: ld.param.u64 %rd{{.*}}, [test1_param_0];
|
|
; CHECK-DEFAULT-32: ld.param.u32 %r{{.*}}, [test1_param_0];
|
|
; CHECK-SHORT-LOCAL: ld.param.u32 %r{{.*}}, [test1_param_0];
|
|
%v = load i8, i8 addrspace(5)* %local
|
|
call void @use(i8 %v)
|
|
ret void
|
|
}
|
|
|
|
define void @test2() {
|
|
%v = alloca i8
|
|
%cast = addrspacecast i8* %v to i8 addrspace(5)*
|
|
; CHECK-DEFAULT: .param .b64 param0;
|
|
; CHECK-DEFAULT: st.param.b64
|
|
; CHECK-DEFAULT-32: .param .b32 param0;
|
|
; CHECK-DEFAULT-32: st.param.b32
|
|
; CHECK-SHORT-LOCAL: .param .b32 param0;
|
|
; CHECK-SHORT-LOCAL: st.param.b32
|
|
call void @test1(i8 addrspace(5)* %cast)
|
|
ret void
|
|
}
|