[WIP][APX] Allow EGPR registers used as non-volatile registers

[phoebewang]

No description provided.

[github-actions]

⚠️ C/C++ code formatter, clang-format found issues in your code. ⚠️

You can test this locally with the following command:

git-clang-format --diff origin/main HEAD --extensions cpp -- clang/lib/Driver/ToolChains/Arch/X86.cpp clang/lib/Driver/ToolChains/CommonArgs.cpp llvm/lib/Target/X86/X86RegisterInfo.cpp --diff_from_common_commit

⚠️
The reproduction instructions above might return results for more than one PR
in a stack if you are using a stacked PR workflow. You can limit the results by
changing origin/main to the base branch/commit you want to compare against.
⚠️

View the diff from clang-format here.

diff --git a/llvm/lib/Target/X86/X86RegisterInfo.cpp b/llvm/lib/Target/X86/X86RegisterInfo.cpp index b22e0ab70..1aadcc39c 100644 --- a/llvm/lib/Target/X86/X86RegisterInfo.cpp +++ b/llvm/lib/Target/X86/X86RegisterInfo.cpp @@ -52,7 +52,8 @@ static cl::opt<bool> extern cl::opt<bool> X86EnableAPXForRelocation; -static cl::opt<unsigned> ApxEgprCSR("apx-egpr-csr", cl::init(0), +static cl::opt<unsigned> ApxEgprCSR( + "apx-egpr-csr", cl::init(0), cl::desc("Set N egpr as callee-saved register for APX calling convention")); X86RegisterInfo::X86RegisterInfo(const Triple &TT) @@ -267,24 +268,42 @@ X86RegisterInfo::getCalleeSavedRegs(const MachineFunction *MF) const { return CSR_Win64_SaveList; switch (ApxEgprCSR) { - case 0: return CSR_Win64_SaveList; - case 1: return CSR_Win64_EGPR1_SaveList; - case 2: return CSR_Win64_EGPR2_SaveList; - case 3: return CSR_Win64_EGPR3_SaveList; - case 4: return CSR_Win64_EGPR4_SaveList; - case 5: return CSR_Win64_EGPR5_SaveList; - case 6: return CSR_Win64_EGPR6_SaveList; - case 7: return CSR_Win64_EGPR7_SaveList; - case 8: return CSR_Win64_EGPR8_SaveList; - case 9: return CSR_Win64_EGPR9_SaveList; - case 10: return CSR_Win64_EGPR10_SaveList; - case 11: return CSR_Win64_EGPR11_SaveList; - case 12: return CSR_Win64_EGPR12_SaveList; - case 13: return CSR_Win64_EGPR13_SaveList; - case 14: return CSR_Win64_EGPR14_SaveList; - case 15: return CSR_Win64_EGPR15_SaveList; - case 16: return CSR_Win64_EGPR16_SaveList; - default: llvm_unreachable("Invalid reg number!"); + case 0: + return CSR_Win64_SaveList; + case 1: + return CSR_Win64_EGPR1_SaveList; + case 2: + return CSR_Win64_EGPR2_SaveList; + case 3: + return CSR_Win64_EGPR3_SaveList; + case 4: + return CSR_Win64_EGPR4_SaveList; + case 5: + return CSR_Win64_EGPR5_SaveList; + case 6: + return CSR_Win64_EGPR6_SaveList; + case 7: + return CSR_Win64_EGPR7_SaveList; + case 8: + return CSR_Win64_EGPR8_SaveList; + case 9: + return CSR_Win64_EGPR9_SaveList; + case 10: + return CSR_Win64_EGPR10_SaveList; + case 11: + return CSR_Win64_EGPR11_SaveList; + case 12: + return CSR_Win64_EGPR12_SaveList; + case 13: + return CSR_Win64_EGPR13_SaveList; + case 14: + return CSR_Win64_EGPR14_SaveList; + case 15: + return CSR_Win64_EGPR15_SaveList; + case 16: + return CSR_Win64_EGPR16_SaveList; + default: + llvm_unreachable("Invalid reg number!"); } }; @@ -423,24 +442,42 @@ X86RegisterInfo::getCallPreservedMask(const MachineFunction &MF, return CSR_Win64_RegMask; switch (ApxEgprCSR) { - case 0: return CSR_Win64_RegMask; - case 1: return CSR_Win64_EGPR1_RegMask; - case 2: return CSR_Win64_EGPR2_RegMask; - case 3: return CSR_Win64_EGPR3_RegMask; - case 4: return CSR_Win64_EGPR4_RegMask; - case 5: return CSR_Win64_EGPR5_RegMask; - case 6: return CSR_Win64_EGPR6_RegMask; - case 7: return CSR_Win64_EGPR7_RegMask; - case 8: return CSR_Win64_EGPR8_RegMask; - case 9: return CSR_Win64_EGPR9_RegMask; - case 10: return CSR_Win64_EGPR10_RegMask; - case 11: return CSR_Win64_EGPR11_RegMask; - case 12: return CSR_Win64_EGPR12_RegMask; - case 13: return CSR_Win64_EGPR13_RegMask; - case 14: return CSR_Win64_EGPR14_RegMask; - case 15: return CSR_Win64_EGPR15_RegMask; - case 16: return CSR_Win64_EGPR16_RegMask; - default: llvm_unreachable("Invalid reg number!"); + case 0: + return CSR_Win64_RegMask; + case 1: + return CSR_Win64_EGPR1_RegMask; + case 2: + return CSR_Win64_EGPR2_RegMask; + case 3: + return CSR_Win64_EGPR3_RegMask; + case 4: + return CSR_Win64_EGPR4_RegMask; + case 5: + return CSR_Win64_EGPR5_RegMask; + case 6: + return CSR_Win64_EGPR6_RegMask; + case 7: + return CSR_Win64_EGPR7_RegMask; + case 8: + return CSR_Win64_EGPR8_RegMask; + case 9: + return CSR_Win64_EGPR9_RegMask; + case 10: + return CSR_Win64_EGPR10_RegMask; + case 11: + return CSR_Win64_EGPR11_RegMask; + case 12: + return CSR_Win64_EGPR12_RegMask; + case 13: + return CSR_Win64_EGPR13_RegMask; + case 14: + return CSR_Win64_EGPR14_RegMask; + case 15: + return CSR_Win64_EGPR15_RegMask; + case 16: + return CSR_Win64_EGPR16_RegMask; + default: + llvm_unreachable("Invalid reg number!"); } };