portelli/Grid
1
0
Fork 0
mirror of https://github.com/paboyle/Grid.git synced 2025-04-04 19:25:56 +01:00
Code Releases Activity

Threading for many core

Browse Source
This commit is contained in:
paboyle 2015-11-06 03:46:41 -08:00
parent 64770d9052
commit b362f8d27b
1 changed files with 12 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
lib/Threads.h
View File

@ -24,7 +24,16 @@ namespace Grid {
class GridThread { class GridThread {
public: public:
static int _threads; static int _threads;
static int _hyperthreads;
static int _cores;
static void SetCores(int cr) {
#ifdef GRID_OMP
_cores = cr;
#else
_cores = 1;
#endif
}
static void SetThreads(int thr) { static void SetThreads(int thr) {
#ifdef GRID_OMP #ifdef GRID_OMP
_threads = MIN(thr,omp_get_max_threads()) ; _threads = MIN(thr,omp_get_max_threads()) ;
@ -35,12 +44,15 @@ class GridThread {
}; };
static void SetMaxThreads(void) { static void SetMaxThreads(void) {
#ifdef GRID_OMP #ifdef GRID_OMP
setenv("KMP_AFFINITY","balanced",1);
_threads = omp_get_max_threads(); _threads = omp_get_max_threads();
omp_set_num_threads(_threads); omp_set_num_threads(_threads);
#else #else
_threads = 1; _threads = 1;
#endif #endif
}; };
static int GetHyperThreads(void) { assert(_threads%_cores ==0); return _threads/_cores; };
static int GetCores(void) { return _cores; };
static int GetThreads(void) { return _threads; }; static int GetThreads(void) { return _threads; };
static int SumArraySize(void) {return _threads;}; static int SumArraySize(void) {return _threads;};