mirror of
https://github.com/RPCS3/rpcs3.git
synced 2025-07-05 22:41:25 +12:00
Atomics: use WaitOnAddress if available (Win8+)
This commit is contained in:
parent
4384ae15b4
commit
f0fd7e2e19
3 changed files with 84 additions and 17 deletions
|
@ -28,6 +28,14 @@ DYNAMIC_IMPORT("ntdll.dll", NtWaitForKeyedEvent, NTSTATUS(HANDLE, PVOID Key, BOO
|
||||||
DYNAMIC_IMPORT("ntdll.dll", NtReleaseKeyedEvent, NTSTATUS(HANDLE, PVOID Key, BOOLEAN Alertable, PLARGE_INTEGER Timeout));
|
DYNAMIC_IMPORT("ntdll.dll", NtReleaseKeyedEvent, NTSTATUS(HANDLE, PVOID Key, BOOLEAN Alertable, PLARGE_INTEGER Timeout));
|
||||||
DYNAMIC_IMPORT("ntdll.dll", NtWaitForSingleObject, NTSTATUS(HANDLE Handle, BOOLEAN Alertable, PLARGE_INTEGER Timeout));
|
DYNAMIC_IMPORT("ntdll.dll", NtWaitForSingleObject, NTSTATUS(HANDLE Handle, BOOLEAN Alertable, PLARGE_INTEGER Timeout));
|
||||||
DYNAMIC_IMPORT("ntdll.dll", NtDelayExecution, NTSTATUS(BOOLEAN Alertable, PLARGE_INTEGER DelayInterval));
|
DYNAMIC_IMPORT("ntdll.dll", NtDelayExecution, NTSTATUS(BOOLEAN Alertable, PLARGE_INTEGER DelayInterval));
|
||||||
|
|
||||||
|
namespace wutex
|
||||||
|
{
|
||||||
|
DYNAMIC_IMPORT("API-MS-Win-Core-Synch-l1-2-0.dll", WaitOnAddress, BOOL(volatile VOID* Address, PVOID CompareAddress, SIZE_T AddressSize, DWORD dwMilliseconds));
|
||||||
|
DYNAMIC_IMPORT("API-MS-Win-Core-Synch-l1-2-0.dll", WakeByAddressSingle, BOOL(PVOID Address));
|
||||||
|
DYNAMIC_IMPORT("API-MS-Win-Core-Synch-l1-2-0.dll", WakeByAddressAll, BOOL(PVOID Address));
|
||||||
|
}
|
||||||
|
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
#ifndef __linux__
|
#ifndef __linux__
|
||||||
|
|
|
@ -818,11 +818,15 @@ bool cpu_thread::suspend_work::push(cpu_thread* _this, bool cancel_if_not_suspen
|
||||||
// Check only CPUs which haven't acknowledged their waiting state yet
|
// Check only CPUs which haven't acknowledged their waiting state yet
|
||||||
for_all_cpu<true>([&](cpu_thread* cpu, u64 index)
|
for_all_cpu<true>([&](cpu_thread* cpu, u64 index)
|
||||||
{
|
{
|
||||||
|
verify(HERE), cpu->state & cpu_flag::pause;
|
||||||
|
|
||||||
if (cpu->state & cpu_flag::wait)
|
if (cpu->state & cpu_flag::wait)
|
||||||
{
|
{
|
||||||
ctr->cpu_copy_bits[index / 64] &= ~(1ull << (index % 64));
|
ctr->cpu_copy_bits[index / 64] &= ~(1ull << (index % 64));
|
||||||
}
|
}
|
||||||
});
|
});
|
||||||
|
|
||||||
|
_mm_pause();
|
||||||
}
|
}
|
||||||
|
|
||||||
// Extract queue and reverse element order (FILO to FIFO) (TODO: maybe leave order as is?)
|
// Extract queue and reverse element order (FILO to FIFO) (TODO: maybe leave order as is?)
|
||||||
|
|
|
@ -214,7 +214,13 @@ namespace
|
||||||
};
|
};
|
||||||
}
|
}
|
||||||
#elif defined(_WIN32)
|
#elif defined(_WIN32)
|
||||||
using sema_handle = std::uint16_t;
|
namespace
|
||||||
|
{
|
||||||
|
struct alignas(64) sema_handle
|
||||||
|
{
|
||||||
|
atomic_t<u64> sema;
|
||||||
|
};
|
||||||
|
}
|
||||||
#else
|
#else
|
||||||
namespace
|
namespace
|
||||||
{
|
{
|
||||||
|
@ -560,6 +566,28 @@ void atomic_storage_futex::wait(const void* data, std::size_t size, u64 old_valu
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
#elif defined(_WIN32) && !defined(USE_POSIX)
|
#elif defined(_WIN32) && !defined(USE_POSIX)
|
||||||
|
if (wutex::WaitOnAddress)
|
||||||
|
{
|
||||||
|
if (s_sema_list[sema_id].sema.try_dec(0))
|
||||||
|
{
|
||||||
|
fallback = true;
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
const DWORD time_ms = timeout + 1 ? INFINITE : (timeout > (UINT32_MAX - 1) * 1000'000 ? (UINT32_MAX - 1) : timeout / 1000'000);
|
||||||
|
|
||||||
|
sema_handle _cmp{};
|
||||||
|
|
||||||
|
wutex::WaitOnAddress(&s_sema_list[sema_id].sema, &_cmp.sema, sizeof(_cmp.sema), time_ms);
|
||||||
|
|
||||||
|
if (s_sema_list[sema_id].sema.try_dec(0))
|
||||||
|
{
|
||||||
|
fallback = true;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
LARGE_INTEGER qw;
|
LARGE_INTEGER qw;
|
||||||
qw.QuadPart = -static_cast<s64>(timeout / 100);
|
qw.QuadPart = -static_cast<s64>(timeout / 100);
|
||||||
|
|
||||||
|
@ -573,6 +601,7 @@ void atomic_storage_futex::wait(const void* data, std::size_t size, u64 old_valu
|
||||||
{
|
{
|
||||||
fallback = true;
|
fallback = true;
|
||||||
}
|
}
|
||||||
|
}
|
||||||
#elif defined(USE_POSIX)
|
#elif defined(USE_POSIX)
|
||||||
struct timespec ts;
|
struct timespec ts;
|
||||||
clock_gettime(CLOCK_REALTIME, &ts);
|
clock_gettime(CLOCK_REALTIME, &ts);
|
||||||
|
@ -671,12 +700,22 @@ void atomic_storage_futex::wait(const void* data, std::size_t size, u64 old_valu
|
||||||
fallback = true;
|
fallback = true;
|
||||||
}
|
}
|
||||||
#elif defined(_WIN32) && !defined(USE_POSIX)
|
#elif defined(_WIN32) && !defined(USE_POSIX)
|
||||||
|
if (wutex::WaitOnAddress)
|
||||||
|
{
|
||||||
|
if (s_sema_list[sema_id].sema.try_dec(0))
|
||||||
|
{
|
||||||
|
fallback = true;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
static LARGE_INTEGER instant{};
|
static LARGE_INTEGER instant{};
|
||||||
|
|
||||||
if (!NtWaitForKeyedEvent(nullptr, &s_sema_list[sema_id], false, &instant))
|
if (!NtWaitForKeyedEvent(nullptr, &s_sema_list[sema_id], false, &instant))
|
||||||
{
|
{
|
||||||
fallback = true;
|
fallback = true;
|
||||||
}
|
}
|
||||||
|
}
|
||||||
#elif defined(USE_POSIX)
|
#elif defined(USE_POSIX)
|
||||||
if (sem_trywait(&s_sema_list[sema_id]) == 0)
|
if (sem_trywait(&s_sema_list[sema_id]) == 0)
|
||||||
{
|
{
|
||||||
|
@ -772,7 +811,15 @@ void atomic_storage_futex::notify_one(const void* data)
|
||||||
s_sema_list[sema_id].sema++;
|
s_sema_list[sema_id].sema++;
|
||||||
futex(&s_sema_list[sema_id].sema, FUTEX_WAKE_PRIVATE, 1);
|
futex(&s_sema_list[sema_id].sema, FUTEX_WAKE_PRIVATE, 1);
|
||||||
#elif defined(_WIN32)
|
#elif defined(_WIN32)
|
||||||
|
if (wutex::WaitOnAddress)
|
||||||
|
{
|
||||||
|
s_sema_list[sema_id].sema++;
|
||||||
|
wutex::WakeByAddressSingle(&s_sema_list[sema_id].sema);
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
NtReleaseKeyedEvent(nullptr, &s_sema_list[sema_id], 1, nullptr);
|
NtReleaseKeyedEvent(nullptr, &s_sema_list[sema_id], 1, nullptr);
|
||||||
|
}
|
||||||
#else
|
#else
|
||||||
dumb_sema& sema = *s_sema_list[sema_id];
|
dumb_sema& sema = *s_sema_list[sema_id];
|
||||||
|
|
||||||
|
@ -830,10 +877,18 @@ void atomic_storage_futex::notify_all(const void* data)
|
||||||
s_sema_list[sema_id].sema += count;
|
s_sema_list[sema_id].sema += count;
|
||||||
futex(&s_sema_list[sema_id].sema, FUTEX_WAKE_PRIVATE, 0x7fff'ffff);
|
futex(&s_sema_list[sema_id].sema, FUTEX_WAKE_PRIVATE, 0x7fff'ffff);
|
||||||
#elif defined(_WIN32)
|
#elif defined(_WIN32)
|
||||||
|
if (wutex::WaitOnAddress)
|
||||||
|
{
|
||||||
|
s_sema_list[sema_id].sema += count;
|
||||||
|
wutex::WakeByAddressAll(&s_sema_list[sema_id].sema);
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
for (u32 i = 0; i < count; i++)
|
for (u32 i = 0; i < count; i++)
|
||||||
{
|
{
|
||||||
NtReleaseKeyedEvent(nullptr, &s_sema_list[sema_id], count, nullptr);
|
NtReleaseKeyedEvent(nullptr, &s_sema_list[sema_id], count, nullptr);
|
||||||
}
|
}
|
||||||
|
}
|
||||||
#else
|
#else
|
||||||
if (count)
|
if (count)
|
||||||
{
|
{
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue