特定のpthreadのCPUアフィニティを指定したいと思います。私がこれまでに見つけたすべてのリファレンスは、スレッド(pthread_t)ではなくプロセス(pid_t)のCPUアフィニティの設定を扱っています。pthread_tを渡していくつかの実験を試みましたが、予想どおり失敗しました。私は不可能なことをしようとしていますか?そうでない場合は、ポインタを送っていただけますか?どうもありがとう。
5 に答える
これは、私の生活を楽にするために私が作ったラッパーです。その効果は、呼び出し元のスレッドが id のコアに「スタック」することcore_id
です。
// core_id = 0, 1, ... n-1, where n is the system's number of cores
int stick_this_thread_to_core(int core_id) {
int num_cores = sysconf(_SC_NPROCESSORS_ONLN);
if (core_id < 0 || core_id >= num_cores)
return EINVAL;
cpu_set_t cpuset;
CPU_ZERO(&cpuset);
CPU_SET(core_id, &cpuset);
pthread_t current_thread = pthread_self();
return pthread_setaffinity_np(current_thread, sizeof(cpu_set_t), &cpuset);
}
Linuxを想定:
アフィニティを設定するためのインターフェイスは、おそらくすでに発見されているように、次のとおりです。
int sched_setaffinity(pid_t pid,size_t cpusetsize,cpu_set_t *mask);
pidとして0を渡すと、現在のスレッドにのみ適用されるか、他のスレッドにLinux固有の呼び出しでカーネルpidを報告させ、それをpidとして渡しますpid_t gettid(void);
。
マニュアルページの引用
アフィニティマスクは、実際にはスレッドごとの属性であり、スレッドグループ内のスレッドごとに個別に調整できます。gettid(2)の呼び出しから返される値は、引数pidで渡すことができます。pidを0として指定すると、呼び出し元のスレッドの属性が設定され、呼び出しから返された値をgetpid(2)に渡すと、スレッドグループのメインスレッドの属性が設定されます。(POSIXスレッドAPIを使用している場合は、sched_setaffinity()の代わりにpthread_setaffinity_np(3)を使用してください。)
//compilation: gcc -o affinity affinity.c -lpthread
#define _GNU_SOURCE
#include <sched.h> //cpu_set_t , CPU_SET
#include <pthread.h> //pthread_t
#include <stdio.h>
void *th_func(void * arg);
int main(void) {
pthread_t thread; //the thread
pthread_create(&thread,NULL,th_func,NULL);
pthread_join(thread,NULL);
return 0;
}
void *th_func(void * arg)
{
//we can set one or more bits here, each one representing a single CPU
cpu_set_t cpuset;
//the CPU we whant to use
int cpu = 2;
CPU_ZERO(&cpuset); //clears the cpuset
CPU_SET( cpu , &cpuset); //set CPU 2 on cpuset
/*
* cpu affinity for the calling thread
* first parameter is the pid, 0 = calling thread
* second parameter is the size of your cpuset
* third param is the cpuset in which your thread will be
* placed. Each bit represents a CPU
*/
sched_setaffinity(0, sizeof(cpuset), &cpuset);
while (1);
; //burns the CPU 2
return 0;
}
POSIX 環境では、cpusets を使用して、プロセスまたは pthread が使用できる CPU を制御できます。このタイプの制御は、CPU アフィニティと呼ばれます。
関数 'sched_setaffinity' は、pthread ID と cpuset をパラメーターとして受け取ります。最初のパラメーターで 0 を使用すると、呼び出し元のスレッドが影響を受けます
Please find the below example program to cpu-affinity of a particular pthread.
Please add appropriate libs.
double waste_time(long n)
{
double res = 0;
long i = 0;
while (i <n * 200000) {
i++;
res += sqrt(i);
}
return res;
}
void *thread_func(void *param)
{
unsigned long mask = 1; /* processor 0 */
/* bind process to processor 0 */
if (pthread_setaffinity_np(pthread_self(), sizeof(mask),
&mask) <0) {
perror("pthread_setaffinity_np");
}
/* waste some time so the work is visible with "top" */
printf("result: %f\n", waste_time(2000));
mask = 2; /* process switches to processor 1 now */
if (pthread_setaffinity_np(pthread_self(), sizeof(mask),
&mask) <0) {
perror("pthread_setaffinity_np");
}
/* waste some more time to see the processor switch */
printf("result: %f\n", waste_time(2000));
}
int main(int argc, char *argv[])
{
pthread_t my_thread;
if (pthread_create(&my_thread, NULL, thread_func, NULL) != 0) {
perror("pthread_create");
}
pthread_exit(NULL);
}
Compile above program with -D_GNU_SOURCE flag.
スケジューラは、CPU アフィニティを必要に応じて変更します。永続的に設定するには、/proc ファイル システムの cpuset を参照してください。
http://man7.org/linux/man-pages/man7/cpuset.7.html
または、sched_setaffinity を使用して定期的に (数秒ごとに) CPU アフィニティを設定する短いプログラムを作成することもできます。