pin_thread_to_core.cpp

Go to the documentation of this file.

 
/***************************************
        Auteur : Pierre Aubert
        Mail : pierre.aubert@lapp.in2p3.fr
        Licence : CeCILL-C
****************************************/
 
#include <errno.h>
#include <stdio.h>
#include <stdlib.h>
#include <pthread.h>
#include <iostream>
#include "pin_thread_to_core.h"
 
#define handle_error_en(en, msg) \
        do { errno = en; perror(msg); exit(EXIT_FAILURE); } while (0)
 
using namespace std;


bool pinThreadToCore(){
#ifndef __APPLE__
        int s, j;
        cpu_set_t cpuset;
        pthread_t thread;
 
        thread = pthread_self();
 
        /* Set affinity mask to include CPUs 0 to 7 */
 
        CPU_ZERO(&cpuset);
        for (j = 0; j < 8; j++)
                CPU_SET(j, &cpuset);
 
        s = pthread_setaffinity_np(thread, sizeof(cpu_set_t), &cpuset);
        if (s != 0){
                handle_error_en(s, "pthread_setaffinity_np");
                return false;
        }
        /* Check the actual affinity mask assigned to the thread */
 
        s = pthread_getaffinity_np(thread, sizeof(cpu_set_t), &cpuset);
        if (s != 0){
                handle_error_en(s, "pthread_getaffinity_np");
                return false;
        }
#endif
        return true;
}