diff options
Diffstat (limited to 'llvm/lib/Target/AMDGPU/SIISelLowering.cpp')
| -rw-r--r-- | llvm/lib/Target/AMDGPU/SIISelLowering.cpp | 10 |
1 files changed, 5 insertions, 5 deletions
diff --git a/llvm/lib/Target/AMDGPU/SIISelLowering.cpp b/llvm/lib/Target/AMDGPU/SIISelLowering.cpp index 6b94333bd41..92c832f7a46 100644 --- a/llvm/lib/Target/AMDGPU/SIISelLowering.cpp +++ b/llvm/lib/Target/AMDGPU/SIISelLowering.cpp @@ -685,9 +685,6 @@ SDValue SITargetLowering::LowerFormalArguments( } if (!AMDGPU::isShader(CallConv)) { - getOriginalFunctionArgs(DAG, DAG.getMachineFunction().getFunction(), Ins, - Splits); - assert(Info->hasWorkGroupIDX() && Info->hasWorkItemIDX()); } else { assert(!Info->hasPrivateSegmentBuffer() && !Info->hasDispatchPtr() && @@ -735,7 +732,10 @@ SDValue SITargetLowering::LowerFormalArguments( CCInfo.AllocateReg(FlatScratchInitReg); } - AnalyzeFormalArguments(CCInfo, Splits); + if (!AMDGPU::isShader(CallConv)) + analyzeFormalArgumentsCompute(CCInfo, Ins); + else + AnalyzeFormalArguments(CCInfo, Splits); SmallVector<SDValue, 16> Chains; @@ -752,7 +752,7 @@ SDValue SITargetLowering::LowerFormalArguments( if (VA.isMemLoc()) { VT = Ins[i].VT; - EVT MemVT = Splits[i].VT; + EVT MemVT = VA.getLocVT(); const unsigned Offset = Subtarget->getExplicitKernelArgOffset() + VA.getLocMemOffset(); // The first 36 bytes of the input buffer contains information about |
