F5/unison
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

mpi: Keep original event UID after automatic partition

Browse Source
This commit is contained in:
F5
2023-11-22 16:29:41 +08:00
parent 4b5f831aa3
commit 86b9cfb981
1 changed files with 14 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
src/mpi/model/hybrid-simulator-impl.cc
View File

@@ -471,22 +471,29 @@ HybridSimulatorImpl::Partition()
<< " threads");
// create new LPs
const Ptr<Scheduler> events = MtpInterface::GetSystem()->GetPendingEvents();
MtpInterface::Disable();
MtpInterface::Enable(threadCount, systemCount);
MtpInterface::EnableNew(threadCount, systemCount);
// set scheduler
ObjectFactory schedulerFactory;
schedulerFactory.SetTypeId(m_schedulerTypeId);
for (uint32_t i = 0; i <= systemCount; i++)
for (uint32_t i = 1; i <= systemCount; i++)
{
MtpInterface::GetSystem(i)->SetScheduler(schedulerFactory);
}
// transfer events to new LPs
while (!events->IsEmpty())
// remove old events in public LP
const Ptr<Scheduler> oldEvents = MtpInterface::GetSystem()->GetPendingEvents();
const Ptr<Scheduler> eventsToBeTransferred = schedulerFactory.Create<Scheduler>();
while (!oldEvents->IsEmpty())
{
Scheduler::Event ev = events->RemoveNext();
Scheduler::Event next = oldEvents->RemoveNext();
eventsToBeTransferred->Insert(next);
}
// transfer events to new LPs
while (!eventsToBeTransferred->IsEmpty())
{
Scheduler::Event ev = eventsToBeTransferred->RemoveNext();
// invoke initialization events (at time 0) by their insertion order
// since changing the execution order of these events may cause error,
// they have to be invoked now rather than parallelly executed