mirror of
https://github.com/wolfpld/tracy.git
synced 2026-10-15 13:25:38 +00:00
The function is about to dispatch between rpmalloc and a pluggable allocator hook, so the rpmalloc-specific name no longer fits. Pure rename plus a small consequence: the SymbolWorker call site no longer needs the TRACY_USE_RPMALLOC guard, since the no-op static-inline fallback in TracyAlloc.hpp makes InitAllocator() safe to call unconditionally.
44 lines
1.0 KiB
C++
44 lines
1.0 KiB
C++
#include "../common/TracyAlloc.hpp"
|
|
|
|
#ifdef TRACY_USE_RPMALLOC
|
|
|
|
#include <atomic>
|
|
|
|
#include "../common/TracyForceInline.hpp"
|
|
#include "../common/TracyYield.hpp"
|
|
|
|
namespace tracy
|
|
{
|
|
|
|
extern thread_local bool RpThreadInitDone;
|
|
extern std::atomic<int> RpInitDone;
|
|
extern std::atomic<int> RpInitLock;
|
|
|
|
tracy_no_inline static void InitAllocatorPlumbing()
|
|
{
|
|
const auto done = RpInitDone.load( std::memory_order_acquire );
|
|
if( !done )
|
|
{
|
|
int expected = 0;
|
|
while( !RpInitLock.compare_exchange_weak( expected, 1, std::memory_order_release, std::memory_order_relaxed ) ) { expected = 0; YieldThread(); }
|
|
const auto done = RpInitDone.load( std::memory_order_acquire );
|
|
if( !done )
|
|
{
|
|
rpmalloc_initialize();
|
|
RpInitDone.store( 1, std::memory_order_release );
|
|
}
|
|
RpInitLock.store( 0, std::memory_order_release );
|
|
}
|
|
rpmalloc_thread_initialize();
|
|
RpThreadInitDone = true;
|
|
}
|
|
|
|
TRACY_API void InitAllocator()
|
|
{
|
|
if( !RpThreadInitDone ) InitAllocatorPlumbing();
|
|
}
|
|
|
|
}
|
|
|
|
#endif
|