[RISCV] Handle big frame

XChy · Jun 28, 2024 · 198be67 · 198be67
1 parent 8e6f257
commit 198be67
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/nnvm/Backend/RISCV/StackAllocator.cpp b/nnvm/Backend/RISCV/StackAllocator.cpp
@@ -139,7 +139,7 @@ void StackAllocator::emitPrologue(LIRBuilder &builder, LIRFunc &func) {
   builder.setInsertPoint(bodyBegin);
   loadRegPlusConstantToReg(builder, builder.phyReg(SP),
                            LIRConst::createInt(-frameSize, LIRValueType::i64),
-                           builder.phyReg(SP), builder.phyReg(A0));
+                           builder.phyReg(SP), builder.phyReg(T0));
 
   for (StackSlot *slot : func.getStackSlots()) {
     if (slot->getType() == StackSlot::CalleeSaved) {
@@ -166,6 +166,6 @@ void StackAllocator::emitEpilogue(LIRBuilder &builder, LIRFunc &func) {
 
     loadRegPlusConstantToReg(builder, builder.phyReg(SP),
                              LIRConst::createInt(frameSize, LIRValueType::i64),
-                             builder.phyReg(SP), builder.phyReg(A0));
+                             builder.phyReg(SP), builder.phyReg(T0));
   }
 }