AuroraSupport/AuroraRuntime
2
1
Fork 0
Code Issues 27 Pull Requests Releases Wiki Activity

[*] Work on AuConditionVariable.NT some more

Browse Source 
[*] Fix high cpu regression in 6af9940b
This commit is contained in:
Reece Wilson 2023-05-30 12:46:38 +01:00
parent b60b580d62
commit 53df1ee81d
2 changed files with 40 additions and 28 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
Source/Async/ThreadPool.cpp
View File

@ -360,30 +360,30 @@ namespace Aurora::Async
{
if (this->sortedWork[i].size())
{
return true;
return false;
}
}
return false;
return true;
}
bool GroupWorkQueue::IsEmpty(AuWorkerId_t id)
{
AU_LOCK_GUARD(this->mutex);
for (AU_ITERATE_N(i, AuAsync::kEWorkPrioCount))
{
for (const auto &[srcId, pA] : this->sortedWork[i])
{
if (id == srcId)
{
return true;
}
}
}
return false;
}
}
}
return true;
}
void GroupWorkQueue::Dequeue(AuList<WorkEntry_t> &queue, int maxPopCount, AuAsync::ThreadId_t id)
{
@ -590,7 +590,6 @@ namespace Aurora::Async
state->pendingWorkItems.clear();
}
// Account for
// while (AuAsync.GetCurrentPool()->runForever());
// in the top most task

39
Source/Threading/Primitives/AuConditionVariable.NT.cpp
View File

@ -70,15 +70,12 @@ namespace Aurora::Threading::Primitives
LARGE_INTEGER word;
word.QuadPart = uTargetTimeNt;
// forced smp stall
// forced smt stall
// see the "hopefully nt is smart enough" comment
if (!bIOU)
{
bIOU = DoTryIf([=]()
{
bool b = true;
return CheckOut(b);
});
bIOU = CheckOut(b);
}
if (bRet)
@ -87,7 +84,7 @@ namespace Aurora::Threading::Primitives
{
auto uNow = this->wlist;
auto waiting = uNow >> 2u;
auto uNext = ((waiting + 1) << 2u) | 1;
auto uNext = ((waiting + 1) << 2u) | (!bool(waiting)) | (uNow & 1);
if (AuAtomicCompareExchange(&this->wlist, uNext, uNow) == uNow)
{
@ -162,13 +159,14 @@ namespace Aurora::Threading::Primitives
}
else
{
bool bIOU {};
while (true)
{
while (true)
{
auto uNow = this->wlist;
auto waiting = uNow >> 2u;
auto uNext = ((waiting + 1) << 2u) | 1;
auto uNext = ((waiting + 1) << 2u) | (!bool(waiting)) | (uNow & 1);
if (AuAtomicCompareExchange(&this->wlist, uNext, uNow) == uNow)
{
@ -177,11 +175,11 @@ namespace Aurora::Threading::Primitives
}
// forced smp stall
bool bIOU = DoTryIf([=]()
if (!bIOU)
{
bool b = true;
return CheckOut(b);
});
bIOU = CheckOut(b);
}
// then into the kernel no matter what (hopefully nt is smart enough to have a fast path)
pNtWaitForKeyedEvent(gKeyedEventHandle, &this->wlist, 0, nullptr);
@ -246,7 +244,19 @@ namespace Aurora::Threading::Primitives
return false;
}
return AuAtomicCompareExchange(&this->signalCount, uSignalNow - 1, uSignalNow) == uSignalNow;
auto uSignalNext = uSignalNow - 1;
if (AuAtomicCompareExchange(&this->signalCount, uSignalNext, uSignalNow) != uSignalNow)
{
return false;
}
if (uSignalNext == 0)
{
InterlockedOr((volatile LONG*)&this->wlist, 1);
}
return true;
});
#endif
}
@ -289,11 +299,13 @@ namespace Aurora::Threading::Primitives
auto uBroadcastIterations = expected;
while (expected && uBroadcastIterations)
{
if (AuAtomicCompareExchange(&this->wlist, ((expected - 1) << 2) /*intentional clear*/, original) == original)
{
AuAtomicAdd(&this->signalCount, 1u);
while (expected && uBroadcastIterations)
{
if (AuAtomicCompareExchange(&this->wlist, ((expected - 1) << 2) /*intentional clear*/, original) == original)
{
pNtReleaseKeyedEvent(gKeyedEventHandle, &this->wlist, FALSE, nullptr);
uBroadcastIterations--;
@ -302,6 +314,7 @@ namespace Aurora::Threading::Primitives
original = this->wlist;
expected = original >> 2;
}
}
#else
::WakeAllConditionVariable(&this->winCond_);
#endif