1
0
Fork 0
mirror of https://git.rwth-aachen.de/acs/public/villas/node/ synced 2025-03-09 00:00:00 +01:00

rt: add kernel::rt::setThreadPriority()

This commit is contained in:
Steffen Vogel 2020-09-13 08:35:27 +02:00
parent 70a616c12d
commit 8ad3ac265b
2 changed files with 38 additions and 18 deletions

View file

@ -28,13 +28,16 @@
#pragma once
#include <pthread.h>
namespace villas {
namespace kernel {
namespace rt {
void init(int priority, int affinity);
void setAffinity(int affinity);
void setProcessAffinity(int affinity);
void setThreadAffinity(pthread_t thread, int affinity);
void setPriority(int priority);

View file

@ -48,7 +48,8 @@ void init(int priority, int affinity)
logger->info("Initialize sub-system");
#ifdef __linux__
int is_rt;
int is_rt, is_isol;
char isolcpus[255];
/* Use FIFO scheduler with real time priority */
is_rt = isPreemptible();
@ -60,8 +61,21 @@ void init(int priority, int affinity)
else
logger->warn("You might want to use the 'priority' setting to increase " PROJECT_NAME "'s process priority");
if (affinity)
setAffinity(affinity);
if (affinity) {
is_isol = get_cmdline_param("isolcpus", isolcpus, sizeof(isolcpus));
if (is_isol)
logger->warn("You should reserve some cores for " PROJECT_NAME " (see 'isolcpus')");
else {
CpuSet cset_pin(affinity);
CpuSet cset_isol(isolcpus);
CpuSet cset_non_isol = ~cset_isol & cset_pin;
if (cset_non_isol.count() > 0)
logger->warn("Affinity setting includes cores which are not isolated: affinity={}, isolcpus={}, non_isolated={}", (std::string) cset_pin, (std::string) cset_isol, (std::string) cset_non_isol);
}
setProcessAffinity(affinity);
}
else
logger->warn("You might want to use the 'affinity' setting to pin " PROJECT_NAME " to dedicate CPU cores");
#else
@ -73,27 +87,15 @@ void init(int priority, int affinity)
}
#ifdef __linux__
void setAffinity(int affinity)
void setProcessAffinity(int affinity)
{
char isolcpus[255];
int is_isol, ret;
int ret;
Logger logger = logging.get("kernel:rt");
/* Pin threads to CPUs by setting the affinity */
CpuSet cset_pin(affinity);
is_isol = get_cmdline_param("isolcpus", isolcpus, sizeof(isolcpus));
if (is_isol)
logger->warn("You should reserve some cores for " PROJECT_NAME " (see 'isolcpus')");
else {
CpuSet cset_isol(isolcpus);
CpuSet cset_non_isol = ~cset_isol & cset_pin;
if (cset_non_isol.count() > 0)
logger->warn("Affinity setting includes cores which are not isolated: affinity={}, isolcpus={}, non_isolated={}", (std::string) cset_pin, (std::string) cset_isol, (std::string) cset_non_isol);
}
ret = sched_setaffinity(0, cset_pin.size(), cset_pin);
if (ret)
throw SystemError("Failed to set CPU affinity to cores: {}", (std::string) cset_pin);
@ -101,6 +103,21 @@ void setAffinity(int affinity)
logger->debug("Set affinity to cores: {}", (std::string) cset_pin);
}
void setThreadAffinity(pthread_t thread, int affinity)
{
int ret;
Logger logger = logging.get("kernel:rt");
CpuSet cset_pin(affinity);
ret = pthread_setaffinity_np(thread, cset_pin.size(), cset_pin);
if (ret)
throw SystemError("Failed to set CPU affinity to cores: {}", (std::string) cset_pin);
logger->debug("Set affinity of thread {} to cores: {}", (long unsigned) thread, (std::string) cset_pin);
}
void setPriority(int priority)
{
int ret;