Reload list info after procedure/suspend

Author: adazem009

src/engine/internal/llvm/instructions/procedures.cpp

@@ -81,6 +81,7 @@ LLVMInstruction *Procedures::buildCallProcedure(LLVMInstruction *ins)
     }
 
     m_utils.reloadVariables(m_utils.targetVariables());
+    m_utils.reloadLists();
     return ins->next;
 }
 

src/engine/internal/llvm/llvmbuildutils.cpp

@@ -329,6 +329,17 @@ void LLVMBuildUtils::reloadVariables(llvm::Value *targetVariables)
     }
 }
 
+void LLVMBuildUtils::reloadLists()
+{
+    // Load list size info
+    if (m_warp) {
+        for (auto &[list, listPtr] : m_listPtrs) {
+            llvm::Value *size = m_builder.CreateLoad(m_builder.getInt64Ty(), listPtr.sizePtr);
+            m_builder.CreateStore(size, listPtr.size);
+        }
+    }
+}
+
 void LLVMBuildUtils::pushScopeLevel()
 {
     m_stringHeap.push_back({});
@@ -1256,6 +1267,7 @@ void LLVMBuildUtils::createSuspend()
         syncVariables(m_targetVariables);
         m_coroutine->createSuspend();
         reloadVariables(m_targetVariables);
+        reloadLists();
 
         if (m_warpArg) {
             m_builder.CreateBr(nextBranch);

src/engine/internal/llvm/llvmbuildutils.h

@@ -63,6 +63,7 @@ class LLVMBuildUtils
 
         void syncVariables(llvm::Value *targetVariables);
         void reloadVariables(llvm::Value *targetVariables);
+        void reloadLists();
 
         void pushScopeLevel();
         void popScopeLevel();