Index: src/compiler/x64/code-generator-x64.cc |
diff --git a/src/compiler/x64/code-generator-x64.cc b/src/compiler/x64/code-generator-x64.cc |
index 13e482d3c1e78df071425e2580e48e5b4d89613e..1458f24309663de36e53667dc40f69ea29d0bbed 100644 |
--- a/src/compiler/x64/code-generator-x64.cc |
+++ b/src/compiler/x64/code-generator-x64.cc |
@@ -1444,14 +1444,31 @@ void CodeGenerator::AssemblePrologue() { |
if (descriptor->kind() == CallDescriptor::kCallAddress) { |
__ pushq(rbp); |
__ movq(rbp, rsp); |
+ int register_save_area_size = 0; |
const RegList saves = descriptor->CalleeSavedRegisters(); |
if (saves != 0) { // Save callee-saved registers. |
- int register_save_area_size = 0; |
for (int i = Register::kNumRegisters - 1; i >= 0; i--) { |
if (!((1 << i) & saves)) continue; |
__ pushq(Register::from_code(i)); |
register_save_area_size += kPointerSize; |
} |
+ } |
+ const RegList saves_fp = descriptor->CalleeSavedFPRegisters(); |
+ if (saves_fp != 0) { // Save callee-saved XMM registers. |
+ const uint32_t saves_fp_count = base::bits::CountPopulation32(saves_fp); |
+ const int stack_size = saves_fp_count * 16; |
+ // Adjust the stack pointer. |
+ __ subp(rsp, Immediate(stack_size)); |
+ // Store the registers on the stack. |
+ int slot_idx = 0; |
+ for (int i = XMMRegister::kMaxNumRegisters - 1; i >= 0; i--) { |
+ if (!((1 << i) & saves_fp)) continue; |
+ __ movdqu(Operand(rsp, 16 * slot_idx), XMMRegister::from_code(i)); |
+ slot_idx++; |
+ } |
+ register_save_area_size += stack_size; |
+ } |
+ if (register_save_area_size > 0) { |
frame()->SetRegisterSaveAreaSize(register_save_area_size); |
} |
} else if (descriptor->IsJSFunctionCall()) { |
@@ -1496,8 +1513,22 @@ void CodeGenerator::AssembleReturn() { |
if (stack_slots > 0) { |
__ addq(rsp, Immediate(stack_slots * kPointerSize)); |
} |
- const RegList saves = descriptor->CalleeSavedRegisters(); |
// Restore registers. |
+ const RegList saves_fp = descriptor->CalleeSavedFPRegisters(); |
+ if (saves_fp != 0) { |
+ const uint32_t saves_fp_count = base::bits::CountPopulation32(saves_fp); |
+ const int stack_size = saves_fp_count * 16; |
+ // Load the registers from the stack. |
+ int slot_idx = 0; |
+ for (int i = 0; i < XMMRegister::kMaxNumRegisters; i++) { |
+ if (!((1 << i) & saves_fp)) continue; |
+ __ movdqu(Operand(rsp, 16 * slot_idx), XMMRegister::from_code(i)); |
paul.l...
2015/06/23 02:15:02
Swap params here, per our slack discussion.
akos.palfi.imgtec
2015/06/23 19:48:57
Done.
|
+ slot_idx++; |
+ } |
+ // Adjust the stack pointer. |
+ __ addp(rsp, Immediate(stack_size)); |
+ } |
+ const RegList saves = descriptor->CalleeSavedRegisters(); |
if (saves != 0) { |
for (int i = 0; i < Register::kNumRegisters; i++) { |
if (!((1 << i) & saves)) continue; |