diff options
| -rw-r--r-- | llvm/lib/Target/AMDGPU/AMDGPUAsmPrinter.cpp | 7 | ||||
| -rw-r--r-- | llvm/test/CodeGen/AMDGPU/call-graph-register-usage.ll | 28 |
2 files changed, 35 insertions, 0 deletions
diff --git a/llvm/lib/Target/AMDGPU/AMDGPUAsmPrinter.cpp b/llvm/lib/Target/AMDGPU/AMDGPUAsmPrinter.cpp index f2d903c8e7b..c4fac3eab78 100644 --- a/llvm/lib/Target/AMDGPU/AMDGPUAsmPrinter.cpp +++ b/llvm/lib/Target/AMDGPU/AMDGPUAsmPrinter.cpp @@ -793,6 +793,7 @@ AMDGPUAsmPrinter::SIFunctionResourceInfo AMDGPUAsmPrinter::analyzeResourceUsage( IsSGPR = false; Width = 3; } else if (AMDGPU::SReg_96RegClass.contains(Reg)) { + IsSGPR = true; Width = 3; } else if (AMDGPU::SReg_128RegClass.contains(Reg)) { assert(!AMDGPU::TTMP_128RegClass.contains(Reg) && @@ -806,6 +807,12 @@ AMDGPUAsmPrinter::SIFunctionResourceInfo AMDGPUAsmPrinter::analyzeResourceUsage( IsSGPR = false; IsAGPR = true; Width = 4; + } else if (AMDGPU::VReg_160RegClass.contains(Reg)) { + IsSGPR = false; + Width = 5; + } else if (AMDGPU::SReg_160RegClass.contains(Reg)) { + IsSGPR = true; + Width = 5; } else if (AMDGPU::SReg_256RegClass.contains(Reg)) { assert(!AMDGPU::TTMP_256RegClass.contains(Reg) && "trap handler registers should not be used"); diff --git a/llvm/test/CodeGen/AMDGPU/call-graph-register-usage.ll b/llvm/test/CodeGen/AMDGPU/call-graph-register-usage.ll index bf17c668713..de008649587 100644 --- a/llvm/test/CodeGen/AMDGPU/call-graph-register-usage.ll +++ b/llvm/test/CodeGen/AMDGPU/call-graph-register-usage.ll @@ -238,6 +238,34 @@ entry: ret void } +; Make sure there's no assert when a sgpr160 is used. +; GCN-LABEL: {{^}}count_use_sgpr160_external_call +; GCN: ; sgpr160 s[{{[0-9]+}}:{{[0-9]+}}] +; CI: NumSgprs: 48 +; VI-NOBUG: NumSgprs: 48 +; VI-BUG: NumSgprs: 96 +; GCN: NumVgprs: 24 +define amdgpu_kernel void @count_use_sgpr160_external_call() { +entry: + tail call void asm sideeffect "; sgpr160 $0", "s"(<5 x i32> <i32 10, i32 11, i32 12, i32 13, i32 14>) #1 + call void @external() + ret void +} + +; Make sure there's no assert when a vgpr160 is used. +; GCN-LABEL: {{^}}count_use_vgpr160_external_call +; GCN: ; vgpr160 v[{{[0-9]+}}:{{[0-9]+}}] +; CI: NumSgprs: 48 +; VI-NOBUG: NumSgprs: 48 +; VI-BUG: NumSgprs: 96 +; GCN: NumVgprs: 24 +define amdgpu_kernel void @count_use_vgpr160_external_call() { +entry: + tail call void asm sideeffect "; vgpr160 $0", "v"(<5 x i32> <i32 10, i32 11, i32 12, i32 13, i32 14>) #1 + call void @external() + ret void +} + attributes #0 = { nounwind noinline norecurse } attributes #1 = { nounwind noinline norecurse } attributes #2 = { nounwind noinline } |
