RPCS3/llvm-mirror
1
0
Fork 0
mirror of https://github.com/RPCS3/llvm-mirror.git synced 2024-11-24 03:33:20 +01:00
Code Issues Projects Releases Wiki Activity

AMDGPU/GlobalISel: Set type on vgpr live in special arguments

Browse Source 
Fixes assertion with workitem ID intrinsics used in non-kernel
functions.

llvm-svn: 371951
This commit is contained in:
Matt Arsenault 2019-09-16 00:33:00 +00:00
parent 57b65f272d
commit 6deb33b319
2 changed files with 29 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
lib/Target/AMDGPU/SIISelLowering.cpp
View File

@ -1671,7 +1671,8 @@ static ArgDescriptor allocateVGPR32Input(CCState &CCInfo, unsigned Mask = ~0u,
assert(Reg != AMDGPU::NoRegister);
MachineFunction &MF = CCInfo.getMachineFunction();
MF.addLiveIn(Reg, &AMDGPU::VGPR_32RegClass);
Register LiveInVReg = MF.addLiveIn(Reg, &AMDGPU::VGPR_32RegClass);
MF.getRegInfo().setType(LiveInVReg, LLT::scalar(32));
return ArgDescriptor::createRegister(Reg, Mask);
}

27
test/CodeGen/AMDGPU/GlobalISel/llvm.amdgcn.workitem.id.ll
View File

@ -88,5 +88,32 @@ bb2:
ret void
}
; ALL-LABEL: {{^}}test_workitem_id_x_func:
; ALL: v_lshrrev_b32_e32 v2, 0, v2
; ALL-NEXT: v_and_b32_e32 v2, 0x3ff, v2
define void @test_workitem_id_x_func(i32 addrspace(1)* %out) #1 {
%id = call i32 @llvm.amdgcn.workitem.id.x()
store i32 %id, i32 addrspace(1)* %out
ret void
}
; ALL-LABEL: {{^}}test_workitem_id_y_func:
; ALL: v_lshrrev_b32_e32 v2, 10, v2
; ALL-NEXT: v_and_b32_e32 v2, 0x3ff, v2
define void @test_workitem_id_y_func(i32 addrspace(1)* %out) #1 {
%id = call i32 @llvm.amdgcn.workitem.id.y()
store i32 %id, i32 addrspace(1)* %out
ret void
}
; ALL-LABEL: {{^}}test_workitem_id_z_func:
; ALL: v_lshrrev_b32_e32 v2, 20, v2
; ALL-NEXT: v_and_b32_e32 v2, 0x3ff, v2
define void @test_workitem_id_z_func(i32 addrspace(1)* %out) #1 {
%id = call i32 @llvm.amdgcn.workitem.id.z()
store i32 %id, i32 addrspace(1)* %out
ret void
}
attributes #0 = { nounwind readnone }
attributes #1 = { nounwind }