[turbofan] Add CalleeSavedFPRegisters to CallDescriptor.

src/compiler/arm64/code-generator-arm64.cc

Index: src/compiler/arm64/code-generator-arm64.cc
diff --git a/src/compiler/arm64/code-generator-arm64.cc b/src/compiler/arm64/code-generator-arm64.cc
index 831627dde5cfed8553a8f036e2b53008a78a3bc1..930411e3b3720c89b87129e2482699ca87995ad8 100644
--- a/src/compiler/arm64/code-generator-arm64.cc
+++ b/src/compiler/arm64/code-generator-arm64.cc
@@ -1077,9 +1077,46 @@ void CodeGenerator::AssemblePrologue() {
     __ SetStackPointer(csp);
     __ Push(lr, fp);
     __ Mov(fp, csp);
-    // TODO(dcarney): correct callee saved registers.
-    __ PushCalleeSavedRegisters();
-    frame()->SetRegisterSaveAreaSize(20 * kPointerSize);
+
+    // Save registers.

[paul.l..., 2015/06/23 02:15:02]

This looks a bit messy to me, but luckily there are arm64 macro-asm function to
do most of this, similar to the way we do on mips with MultiPush/Pop. I'll send
a complete diff separately, but the basic save code can look like this:

    CPURegList callee_save_fp =
        CPURegList(CPURegister::kFPRegister, kDRegSizeInBits,
                   descriptor->CalleeSavedFPRegisters());
    DCHECK(callee_save_fp.list() == CPURegList::GetCalleeSavedFP().list());
    int saved_count = callee_save_fp.Count();
    __ PushCPURegList(callee_save_fp);

    CPURegList callee_save = CPURegList(CPURegister::kRegister, kXRegSizeInBits,
                                        descriptor->CalleeSavedRegisters());
    // TODO(): TF save list not in sync with CPURegList::GetCalleeSaved() - x30
    // DCHECK(callee_save.list() == CPURegList::GetCalleeSaved().list());
    saved_count += callee_save.Count();
    __ PushCPURegList(callee_save);

    frame()->SetRegisterSaveAreaSize(saved_count * kPointerSize);

the 2nd DCHECK is a problem since the original TF code saves link-reg (X30), but
the assembler CPURegList::GetCalleeSaved() does not. I have a workaround I'll
send separately, but when we publish this, the arm team can help get the best
approach.

I don't think there is any perf issue here, as their Push/Pop macros do convert
into stp/ldp.

[akos.palfi.imgtec, 2015/06/23 19:48:57]

Thanks, it's much cleaner now.

+    MemOperand tos(csp, -2 * static_cast<int>(kXRegSize), PreIndex);
+    int saved_count = 0;
+    const RegList saves_fp = descriptor->CalleeSavedFPRegisters();
+    if (saves_fp != 0) {
+      // Save callee-saved FP registers.
+      int prev_reg = -1;
+      for (int i = FPRegister::kMaxNumRegisters - 1; i >= 0; i--) {
+        if (((1 << i) & saves_fp)) {
+          if (prev_reg != -1) {
+            __ stp(FPRegister::from_code(i), FPRegister::from_code(prev_reg),
+                   tos);
+            saved_count += 2;
+            prev_reg = -1;
+          } else {
+            prev_reg = i;
+          }
+        }
+      }
+      CHECK(prev_reg == -1);
+    }
+    const RegList saves = descriptor->CalleeSavedRegisters();
+    if (saves != 0) {
+      // Save callee-saved registers.
+      int prev_reg = -1;
+      for (int i = Register::kNumRegisters - 1; i >= 0; i--) {
+        if (((1 << i) & saves)) {
+          if (prev_reg != -1) {
+            __ stp(Register::from_code(i), Register::from_code(prev_reg), tos);
+            saved_count += 2;
+            prev_reg = -1;
+          } else {
+            prev_reg = i;
+          }
+        }
+      }
+      CHECK(prev_reg == -1);
+    }
+    frame()->SetRegisterSaveAreaSize(saved_count * kPointerSize);
   } else if (descriptor->IsJSFunctionCall()) {
     CompilationInfo* info = this->info();
     __ SetStackPointer(jssp);
@@ -1128,10 +1165,45 @@ void CodeGenerator::AssembleReturn() {
       if (stack_slots > 0) {
         __ Add(csp, csp, AlignedStackSlots(stack_slots) * kPointerSize);
       }
+
       // Restore registers.
-      // TODO(dcarney): correct callee saved registers.
-      __ PopCalleeSavedRegisters();
+      MemOperand tos(csp, 2 * kXRegSize, PostIndex);
+      const RegList saves = descriptor->CalleeSavedRegisters();
+      if (saves != 0) {
+        // Restore callee-saved registers.
+        int prev_reg = -1;
+        for (int i = 0; i < Register::kNumRegisters; i++) {
+          if (((1 << i) & saves)) {
+            if (prev_reg != -1) {
+              __ ldp(Register::from_code(prev_reg), Register::from_code(i),
+                     tos);
+              prev_reg = -1;
+            } else {
+              prev_reg = i;
+            }
+          }
+        }
+        CHECK(prev_reg == -1);
+      }
+      const RegList saves_fp = descriptor->CalleeSavedFPRegisters();
+      if (saves_fp != 0) {
+        // Restore callee-saved FP registers.
+        int prev_reg = -1;
+        for (int i = 0; i < FPRegister::kMaxNumRegisters; i++) {
+          if (((1 << i) & saves_fp)) {
+            if (prev_reg != -1) {
+              __ ldp(FPRegister::from_code(prev_reg), FPRegister::from_code(i),
+                     tos);
+              prev_reg = -1;
+            } else {
+              prev_reg = i;
+            }
+          }
+        }
+        CHECK(prev_reg == -1);
+      }
     }
+
     __ Mov(csp, fp);
     __ Pop(fp, lr);
     __ Ret();