commit
77f809e8db
@ -0,0 +1,117 @@
|
||||
/*
|
||||
Copyright (c) 2019, jtgrassie
|
||||
Copyright (c) 2019, tevador <tevador@gmail.com>
|
||||
|
||||
All rights reserved.
|
||||
|
||||
Redistribution and use in source and binary forms, with or without
|
||||
modification, are permitted provided that the following conditions are met:
|
||||
* Redistributions of source code must retain the above copyright
|
||||
notice, this list of conditions and the following disclaimer.
|
||||
* Redistributions in binary form must reproduce the above copyright
|
||||
notice, this list of conditions and the following disclaimer in the
|
||||
documentation and/or other materials provided with the distribution.
|
||||
* Neither the name of the copyright holder nor the
|
||||
names of its contributors may be used to endorse or promote products
|
||||
derived from this software without specific prior written permission.
|
||||
|
||||
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
|
||||
ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
|
||||
WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
|
||||
DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE
|
||||
FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
|
||||
DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
|
||||
SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
|
||||
CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
|
||||
OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
|
||||
OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
||||
*/
|
||||
|
||||
#include <sstream>
|
||||
|
||||
#if defined(_WIN32) || defined(__CYGWIN__)
|
||||
#include <windows.h>
|
||||
#else
|
||||
#ifdef __APPLE__
|
||||
#include <mach/thread_act.h>
|
||||
#include <mach/thread_policy.h>
|
||||
#endif
|
||||
#include <pthread.h>
|
||||
#endif
|
||||
#include "affinity.hpp"
|
||||
|
||||
int
|
||||
set_thread_affinity(const unsigned &cpuid)
|
||||
{
|
||||
std::thread::native_handle_type thread;
|
||||
#if defined(_WIN32) || defined(__CYGWIN__)
|
||||
thread = static_cast<std::thread::native_handle_type>(GetCurrentThread());
|
||||
#else
|
||||
thread = static_cast<std::thread::native_handle_type>(pthread_self());
|
||||
#endif
|
||||
return set_thread_affinity(thread, cpuid);
|
||||
}
|
||||
|
||||
int
|
||||
set_thread_affinity(std::thread::native_handle_type thread,
|
||||
const unsigned &cpuid)
|
||||
{
|
||||
int rc = -1;
|
||||
#ifdef __APPLE__
|
||||
thread_port_t mach_thread;
|
||||
thread_affinity_policy_data_t policy = { static_cast<integer_t>(cpuid) };
|
||||
mach_thread = pthread_mach_thread_np(thread);
|
||||
rc = thread_policy_set(mach_thread, THREAD_AFFINITY_POLICY,
|
||||
(thread_policy_t)&policy, 1);
|
||||
#elif defined(_WIN32) || defined(__CYGWIN__)
|
||||
rc = SetThreadAffinityMask(thread, 1ULL << cpuid) == 0 ? -2 : 0;
|
||||
#else
|
||||
cpu_set_t cs;
|
||||
CPU_ZERO(&cs);
|
||||
CPU_SET(cpuid, &cs);
|
||||
rc = pthread_setaffinity_np(thread, sizeof(cpu_set_t), &cs);
|
||||
#endif
|
||||
return rc;
|
||||
}
|
||||
|
||||
unsigned
|
||||
cpuid_from_mask(uint64_t mask, const unsigned &thread_index)
|
||||
{
|
||||
static unsigned lookup[64];
|
||||
static bool init = false;
|
||||
if (init)
|
||||
return lookup[thread_index];
|
||||
unsigned count_found = 0;
|
||||
for (unsigned i=0; i<64; i++)
|
||||
{
|
||||
if (1ULL & mask)
|
||||
{
|
||||
lookup[count_found] = i;
|
||||
count_found++;
|
||||
}
|
||||
mask >>= 1;
|
||||
}
|
||||
init = true;
|
||||
return lookup[thread_index];
|
||||
}
|
||||
|
||||
std::string
|
||||
mask_to_string(uint64_t mask)
|
||||
{
|
||||
std::ostringstream ss;
|
||||
unsigned len = 0;
|
||||
unsigned v = 0;
|
||||
unsigned i = 64;
|
||||
while (i--)
|
||||
{
|
||||
v = mask >> i;
|
||||
if (1ULL & v)
|
||||
{
|
||||
if (len == 0) len = i + 1;
|
||||
ss << '1';
|
||||
}
|
||||
else
|
||||
if (len > 0) ss << '0';
|
||||
}
|
||||
return ss.str();
|
||||
}
|
@ -0,0 +1,39 @@
|
||||
/*
|
||||
Copyright (c) 2019, jtgrassie
|
||||
|
||||
All rights reserved.
|
||||
|
||||
Redistribution and use in source and binary forms, with or without
|
||||
modification, are permitted provided that the following conditions are met:
|
||||
* Redistributions of source code must retain the above copyright
|
||||
notice, this list of conditions and the following disclaimer.
|
||||
* Redistributions in binary form must reproduce the above copyright
|
||||
notice, this list of conditions and the following disclaimer in the
|
||||
documentation and/or other materials provided with the distribution.
|
||||
* Neither the name of the copyright holder nor the
|
||||
names of its contributors may be used to endorse or promote products
|
||||
derived from this software without specific prior written permission.
|
||||
|
||||
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
|
||||
ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
|
||||
WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
|
||||
DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE
|
||||
FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
|
||||
DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
|
||||
SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
|
||||
CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
|
||||
OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
|
||||
OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
||||
*/
|
||||
|
||||
#pragma once
|
||||
|
||||
#include <cstddef>
|
||||
#include <thread>
|
||||
#include <string>
|
||||
|
||||
int set_thread_affinity(const unsigned &cpuid);
|
||||
int set_thread_affinity(std::thread::native_handle_type thread,
|
||||
const unsigned &cpuid);
|
||||
unsigned cpuid_from_mask(uint64_t mask, const unsigned &thread_index);
|
||||
std::string mask_to_string(uint64_t mask);
|
Loading…
Reference in new issue