[SubZero] Fix size of arguments on stack

src/IceTargetLoweringMIPS32.cpp

 //
 //                        The Subzero Code Generator
 //
 // This file is distributed under the University of Illinois Open Source
 // License. See LICENSE.TXT for details.
 //
 //===----------------------------------------------------------------------===//
 ///
 /// \file
 /// \brief Implements the TargetLoweringMIPS32 class, which consists almost
@@ skipping 207 matching lines @@
 }
 
 uint32_t TargetMIPS32::getStackAlignment() const {
   return MIPS32_STACK_ALIGNMENT_BYTES;
 }
 
 uint32_t TargetMIPS32::getCallStackArgumentsSizeBytes(const InstCall *Call) {
   TargetMIPS32::CallingConv CC;
   RegNumT DummyReg;
   size_t OutArgsSizeBytes = 0;
+  Variable *Dest = Call->getDest();
+  bool PartialOnStack = false;
+  if (Dest != nullptr && isVectorFloatingType(Dest->getType())) {
+    CC.discardReg(RegMIPS32::Reg_A0);
+    // Next vector is partially on stack
+    PartialOnStack = true;
+  }
   for (SizeT i = 0, NumArgs = Call->getNumArgs(); i < NumArgs; ++i) {
     Operand *Arg = legalizeUndef(Call->getArg(i));
     const Type Ty = Arg->getType();
     RegNumT RegNum;
     if (CC.argInReg(Ty, i, &RegNum)) {
+      // If PartialOnStack is true and if this is a vector type then last two
+      // elements are on stack
+      if (PartialOnStack && isVectorType(Ty)) {
+        OutArgsSizeBytes = applyStackAlignmentTy(OutArgsSizeBytes, IceType_i32);
+        OutArgsSizeBytes += typeWidthInBytesOnStack(IceType_i32) * 2;
+      }
       continue;
     }
-
     OutArgsSizeBytes = applyStackAlignmentTy(OutArgsSizeBytes, Ty);
     OutArgsSizeBytes += typeWidthInBytesOnStack(Ty);
   }
-
+  // Add size of argument save area
+  constexpr int BytesPerStackArg = 4;
+  OutArgsSizeBytes += MIPS32_MAX_GPR_ARG * BytesPerStackArg;
   return applyStackAlignment(OutArgsSizeBytes);
 }
 
 namespace {
 inline uint64_t getConstantMemoryOrder(Operand *Opnd) {
   if (auto *Integer = llvm::dyn_cast<ConstantInteger32>(Opnd))
     return Integer->getValue();
   return Intrinsics::MemoryOrderInvalid;
 }
 }
@@ skipping 5712 matching lines @@
     if (auto *CallTargetR = llvm::dyn_cast<Variable>(CallTarget)) {
       Variable *T6 = Target->getPhysicalRegister(RegMIPS32::Reg_T6);
       Target->_and(CallTargetR, CallTargetR, T6);
     }
   }
   return Target->Context.insert<InstMIPS32Call>(ReturnReg, CallTarget);
 }
 
 } // end of namespace MIPS32
 } // end of namespace Ice