1 /*
2 * Copyright (c) 2021-2023 Huawei Device Co., Ltd.
3 * Licensed under the Apache License, Version 2.0 (the "License");
4 * you may not use this file except in compliance with the License.
5 * You may obtain a copy of the License at
6 *
7 * http://www.apache.org/licenses/LICENSE-2.0
8 *
9 * Unless required by applicable law or agreed to in writing, software
10 * distributed under the License is distributed on an "AS IS" BASIS,
11 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12 * See the License for the specific language governing permissions and
13 * limitations under the License.
14 */
15
16 /*
17 Register file implementation.
18 Reserve registers.
19 */
20
21 #include "registers_description.h"
22 #include "target/amd64/target.h"
23 #include "regfile.h"
24
25 namespace panda::compiler::amd64 {
Amd64RegisterDescription(ArenaAllocator * allocator)26 Amd64RegisterDescription::Amd64RegisterDescription(ArenaAllocator *allocator)
27 : RegistersDescription(allocator, Arch::X86_64), usedRegs_(allocator->Adapter())
28 {
29 }
30
IsRegUsed(ArenaVector<Reg> vecReg,Reg reg)31 bool Amd64RegisterDescription::IsRegUsed(ArenaVector<Reg> vecReg, Reg reg)
32 {
33 auto equality = [reg](Reg in) { return (reg.GetId() == in.GetId()) && (reg.GetType() == in.GetType()); };
34 return (std::find_if(vecReg.begin(), vecReg.end(), equality) != vecReg.end());
35 }
36
GetCalleeSaved()37 ArenaVector<Reg> Amd64RegisterDescription::GetCalleeSaved()
38 {
39 ArenaVector<Reg> out(GetAllocator()->Adapter());
40 for (uint32_t i = 0; i < MAX_NUM_REGS; ++i) {
41 if (calleeSaved_.Has(i)) {
42 out.emplace_back(Reg(i, INT64_TYPE));
43 }
44 if (calleeSavedv_.Has(i)) {
45 out.emplace_back(Reg(i, FLOAT64_TYPE));
46 }
47 }
48 return out;
49 }
50
SetCalleeSaved(const ArenaVector<Reg> & regs)51 void Amd64RegisterDescription::SetCalleeSaved(const ArenaVector<Reg> ®s)
52 {
53 calleeSaved_ = RegList(GetCalleeRegsMask(Arch::X86_64, false).GetValue());
54 calleeSavedv_ = RegList(GetCalleeRegsMask(Arch::X86_64, true).GetValue()); // empty
55
56 for (uint32_t i = 0; i < MAX_NUM_REGS; ++i) {
57 bool scalarUsed = IsRegUsed(regs, Reg(i, INT64_TYPE));
58 if (scalarUsed) {
59 calleeSaved_.Add(i);
60 } else {
61 calleeSaved_.Remove(i);
62 }
63 bool vectorUsed = IsRegUsed(regs, Reg(i, FLOAT64_TYPE));
64 if (vectorUsed) {
65 calleeSavedv_.Add(i);
66 } else {
67 calleeSavedv_.Remove(i);
68 }
69 }
70 // Remove return-value from callee
71 calleeSaved_.Remove(ConvertRegNumber(asmjit::x86::rax.id()));
72 }
73
SetUsedRegs(const ArenaVector<Reg> & regs)74 void Amd64RegisterDescription::SetUsedRegs(const ArenaVector<Reg> ®s)
75 {
76 usedRegs_ = regs;
77
78 // Update current lists - to do not use old data
79 calleeSaved_ = RegList(GetCalleeRegsMask(Arch::X86_64, false).GetValue());
80 callerSaved_ = RegList(GetCallerRegsMask(Arch::X86_64, false).GetValue());
81
82 calleeSavedv_ = RegList(GetCalleeRegsMask(Arch::X86_64, true).GetValue()); // empty
83 callerSavedv_ = RegList(GetCallerRegsMask(Arch::X86_64, true).GetValue());
84
85 for (uint32_t i = 0; i < MAX_NUM_REGS; ++i) {
86 // IsRegUsed use used_regs_ variable
87 bool scalarUsed = IsRegUsed(usedRegs_, Reg(i, INT64_TYPE));
88 if (!scalarUsed && calleeSaved_.Has(i)) {
89 calleeSaved_.Remove(i);
90 }
91 if (!scalarUsed && callerSaved_.Has(i)) {
92 callerSaved_.Remove(i);
93 }
94
95 bool vectorUsed = IsRegUsed(usedRegs_, Reg(i, FLOAT64_TYPE));
96 if (!vectorUsed && calleeSavedv_.Has(i)) {
97 calleeSavedv_.Remove(i);
98 }
99 if (!vectorUsed && callerSavedv_.Has(i)) {
100 callerSavedv_.Remove(i);
101 }
102 }
103 }
104
105 } // namespace panda::compiler::amd64
106