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 <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)


bool pinThreadToCore(){
#ifndef __APPLE__
        cpu_set_t cpuset;
        pthread_t thread = pthread_self();
        /* Set affinity mask to include CPUs 0 to 7 */
 
        CPU_ZERO(&cpuset);
        for(int  j = 0; j < 8; j++){
                CPU_SET(j, &cpuset);
        }
        int 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;
}