Merge pull request #3563 from Sonicadvance1/fill_spill_pairs

JIT: Adds support for spilling/Filling GPRPair
FEX-Emu · Apr 9, 2024 · eedb120 · eedb120
2 parents 1616b4e + 574fdce
commit eedb120
Showing 1 changed file with 58 additions and 1 deletion.
diff --git a/FEXCore/Source/Interface/Core/JIT/Arm64/MemoryOps.cpp b/FEXCore/Source/Interface/Core/JIT/Arm64/MemoryOps.cpp
@@ -771,7 +771,36 @@ DEF_OP(SpillRegister) {
       LOGMAN_MSG_A_FMT("Unhandled SpillRegister size: {}", OpSize);
       break;
     }
-  } else {
+  } else if (Op->Class == FEXCore::IR::GPRPairClass) {
+    const auto Src = GetRegPair(Op->Value.ID());
+    switch (OpSize) {
+      case 8: {
+        if (SlotOffset <= 252 && (SlotOffset & 0b11) == 0) {
+          stp<ARMEmitter::IndexType::OFFSET>(Src.first.W(), Src.second.W(), ARMEmitter::Reg::rsp, SlotOffset);
+        }
+        else {
+          add(ARMEmitter::Size::i64Bit, TMP1, ARMEmitter::Reg::rsp, SlotOffset);
+          stp<ARMEmitter::IndexType::OFFSET>(Src.first.W(), Src.second.W(), TMP1, 0);
+        }
+        break;
+      }
+
+      case 16: {
+        if (SlotOffset <= 504 && (SlotOffset & 0b111) == 0) {
+          stp<ARMEmitter::IndexType::OFFSET>(Src.first.X(), Src.second.X(), ARMEmitter::Reg::rsp, SlotOffset);
+        }
+        else {
+          add(ARMEmitter::Size::i64Bit, TMP1, ARMEmitter::Reg::rsp, SlotOffset);
+          stp<ARMEmitter::IndexType::OFFSET>(Src.first.X(), Src.second.X(), TMP1, 0);
+        }
+        break;
+      }
+      default:
+        LOGMAN_MSG_A_FMT("Unhandled SpillRegister(GPRPair) size: {}", OpSize);
+        break;
+    }
+  }
+  else {
     LOGMAN_MSG_A_FMT("Unhandled SpillRegister class: {}", Op->Class.Val);
   }
 }
@@ -871,6 +900,34 @@ DEF_OP(FillRegister) {
       LOGMAN_MSG_A_FMT("Unhandled FillRegister size: {}", OpSize);
       break;
     }
+  } else if (Op->Class == FEXCore::IR::GPRPairClass) {
+    const auto Src = GetRegPair(Node);
+    switch (OpSize) {
+      case 8: {
+        if (SlotOffset <= 252 && (SlotOffset & 0b11) == 0) {
+          ldp<ARMEmitter::IndexType::OFFSET>(Src.first.W(), Src.second.W(), ARMEmitter::Reg::rsp, SlotOffset);
+        }
+        else {
+          add(ARMEmitter::Size::i64Bit, TMP1, ARMEmitter::Reg::rsp, SlotOffset);
+          ldp<ARMEmitter::IndexType::OFFSET>(Src.first.W(), Src.second.W(), TMP1, 0);
+        }
+        break;
+      }
+
+      case 16: {
+        if (SlotOffset <= 504 && (SlotOffset & 0b111) == 0) {
+          ldp<ARMEmitter::IndexType::OFFSET>(Src.first.X(), Src.second.X(), ARMEmitter::Reg::rsp, SlotOffset);
+        }
+        else {
+          add(ARMEmitter::Size::i64Bit, TMP1, ARMEmitter::Reg::rsp, SlotOffset);
+          ldp<ARMEmitter::IndexType::OFFSET>(Src.first.X(), Src.second.X(), TMP1, 0);
+        }
+        break;
+      }
+      default:
+        LOGMAN_MSG_A_FMT("Unhandled FillRegister(GPRPair) size: {}", OpSize);
+        break;
+    }
   } else {
     LOGMAN_MSG_A_FMT("Unhandled FillRegister class: {}", Op->Class.Val);
   }