thread.h

Go to the documentation of this file.

#pragma once

#include <stdbool.h>
#include <arch/atomic.h>

#if defined(OS_LINUX)

#include <sched.h>
#include <unistd.h>
#include <pthread.h>

#define get_cores() (sysconf( _SC_NPROCESSORS_ONLN ))

typedef pthread_t tid_t;

#define new_thread(entry, arg)  pthread_create(&os_tid, NULL, entry, arg)

static inline void set_affinity(int core)
{
    cpu_set_t cpuset;
    CPU_ZERO(&cpuset);
    CPU_SET(core, &cpuset);
    // 0 is the current thread
    sched_setaffinity(0, sizeof(cpuset), &cpuset);
}

#elif defined(OS_WINDOWS)

#include <windows.h>

#define get_cores() ({\
            SYSTEM_INFO _sysinfo;\
            GetSystemInfo( &_sysinfo );\
            _sysinfo.dwNumberOfProcessors;\
            })

typedef HANDLE tid_t;

#define new_thread(entry, arg)  CreateThread(NULL, 0, entry, arg, 0, &os_tid)

#define set_affinity(core) SetThreadAffinityMask(GetCurrentThread(), 1<<core)

#else /* OS_LINUX || OS_WINDOWS */
#error Unsupported operating system
#endif


struct _helper_thread {
    void *(*start_routine)(void *); 
    void *arg;          
};

typedef struct {
    int num_threads;    
    atomic_t c1;        
    atomic_t c2;        
    atomic_t barr;      
} barrier_t;

#define HALF_UINT_BITS (sizeof(unsigned int) * 8 / 2)


#define MAX_KERNELS ((1 << HALF_UINT_BITS) - 1)


#define MAX_THREADS_PER_KERNEL ((1 << HALF_UINT_BITS) - 1)


#define to_global_tid(kid, local_tid) ( (kid << HALF_UINT_BITS) | local_tid )

#define master_thread() (local_tid == 0)

#define thread_barrier_reset(b)     do { \
                        (atomic_set((&b->c1), (b)->num_threads)); \
                        (atomic_set((&b->c2), (b)->num_threads)); \
                        (atomic_set((&b->barr), -1)); \
                    } while (0)


extern __thread unsigned int tid;
extern __thread unsigned int local_tid;

extern void barrier_init(barrier_t * b, int t);
extern bool thread_barrier(barrier_t * b);
extern void create_threads(unsigned short int n, void *(*start_routine)(void *), void *arg);