Fix arg_buffer under Generate_WasmFXSuspend

Also change the tmp register passed to LoadJumpBuffer to avoid
overwriting arg_buffer.

Change-Id: I72b83edf722481ca7018731d36e1939620612e97
Reviewed-on: https://chromium-review.googlesource.com/c/v8/v8/+/7212934
Reviewed-by: John <junyan1@ibm.com>
Commit-Queue: Milad Farazmand <mfarazma@ibm.com>
Cr-Commit-Position: refs/heads/main@{#104032}

Author: miladfarca

src/builtins/ppc/builtins-ppc.cc

@@ -3558,7 +3558,7 @@ void Builtins::Generate_WasmFXSuspend(MacroAssembler* masm) {
   __ EnterFrame(StackFrame::WASM_STACK_EXIT);
   Register tag = WasmFXSuspendDescriptor::GetRegisterParameter(0);
   Register cont = WasmFXSuspendDescriptor::GetRegisterParameter(1);
-  Register arg_buffer = WasmFXResumeDescriptor::GetRegisterParameter(1);
+  Register arg_buffer = WasmFXSuspendDescriptor::GetRegisterParameter(2);
   Label resume;
   __ Push(arg_buffer);
   __ Push(cont, kContextRegister);
@@ -3587,8 +3587,8 @@ void Builtins::Generate_WasmFXSuspend(MacroAssembler* masm) {
 
   __ bind(&ok);
   DCHECK_EQ(cont, kReturnRegister0);
-  DCHECK(!AreAliased(r6, arg_buffer, target_stack));
-  LoadJumpBuffer(masm, target_stack, true, r6);
+  DCHECK(!AreAliased(ip, arg_buffer, target_stack));
+  LoadJumpBuffer(masm, target_stack, true, ip);
   __ Trap();
   __ bind(&resume);
   __ Move(kReturnRegister0, WasmFXResumeDescriptor::GetRegisterParameter(1));

src/builtins/s390/builtins-s390.cc

@@ -3565,7 +3565,7 @@ void Builtins::Generate_WasmFXSuspend(MacroAssembler* masm) {
   __ EnterFrame(StackFrame::WASM_STACK_EXIT);
   Register tag = WasmFXSuspendDescriptor::GetRegisterParameter(0);
   Register cont = WasmFXSuspendDescriptor::GetRegisterParameter(1);
-  Register arg_buffer = WasmFXResumeDescriptor::GetRegisterParameter(1);
+  Register arg_buffer = WasmFXSuspendDescriptor::GetRegisterParameter(2);
   Label resume;
   __ Push(arg_buffer);
   __ Push(cont, kContextRegister);
@@ -3594,8 +3594,8 @@ void Builtins::Generate_WasmFXSuspend(MacroAssembler* masm) {
 
   __ bind(&ok);
   DCHECK_EQ(cont, kReturnRegister0);
-  DCHECK(!AreAliased(r5, arg_buffer, target_stack));
-  LoadJumpBuffer(masm, target_stack, true, r5);
+  DCHECK(!AreAliased(r1, arg_buffer, target_stack));
+  LoadJumpBuffer(masm, target_stack, true, r1);
   __ Trap();
   __ bind(&resume);
   __ Move(kReturnRegister0, WasmFXResumeDescriptor::GetRegisterParameter(1));