diff --git a/core/misc/platform_thread.h b/core/misc/platform_thread.h index 556bac1..d495ab9 100644 --- a/core/misc/platform_thread.h +++ b/core/misc/platform_thread.h @@ -7,6 +7,8 @@ #include #elif PLATFORM_MACOS #include +#include +#include #endif inline void set_thread_name(const char* name) { @@ -23,10 +25,24 @@ inline void set_thread_affinity(std::thread& thread, int core) { #if PLATFORM_WINDOWS const auto handle = reinterpret_cast(thread.native_handle()); SetThreadAffinityMask(handle, 1 << core); -#elif PLATFORM_LINUX || PLATFORM_MACOS +#elif PLATFORM_LINUX cpu_set_t cpuset; CPU_ZERO(&cpuset); CPU_SET(core, &cpuset); pthread_setaffinity_np(thread.native_handle(), sizeof(cpu_set_t), &cpuset); +#elif PLATFORM_MACOS + thread_affinity_policy_data_t policy = { core }; + kern_return_t result = thread_policy_set( + pthread_mach_thread_np(thread.native_handle()), + THREAD_AFFINITY_POLICY, + (thread_policy_t)&policy, + THREAD_AFFINITY_POLICY_COUNT + ); + + if (result != KERN_SUCCESS) { + spdlog::error("Failed to set thread affinity: {}", result); + } else { + spdlog::info("Thread bound to core {}", core); + } #endif }