From 70462345751be91c39abaa0a370ffec2bcc52467 Mon Sep 17 00:00:00 2001
From: Berenger Bramas <Berenger.Bramas@inria.fr>
Date: Fri, 2 Oct 2015 13:26:52 +0200
Subject: [PATCH] update binding functions

---
 Src/Utils/FBinding.hpp | 14 +++++++++++---
 1 file changed, 11 insertions(+), 3 deletions(-)

diff --git a/Src/Utils/FBinding.hpp b/Src/Utils/FBinding.hpp
index f04df125..09f0d91a 100644
--- a/Src/Utils/FBinding.hpp
+++ b/Src/Utils/FBinding.hpp
@@ -17,15 +17,23 @@ namespace FBinding {
 #define FBINDING_ENABLE
 #endif
 
-inline int GetThreadBinding(){
+inline long GetThreadBinding(){
     // Mask will contain the current affinity
 #ifdef FBINDING_ENABLE
-    unsigned long mask = 0;
+    cpu_set_t mask;
+    CPU_ZERO(&mask);
     // We need the thread pid (even if we are in openmp)
     pid_t tid = (pid_t) syscall(SYS_gettid);
     // Get the affinity
     FAssertLF(sched_getaffinity(tid, sizeof(mask), (cpu_set_t*)&mask) != -1);
-    return int(mask>>1);
+
+    long cpus = 0;
+    for(size_t idxCpu = 0 ; idxCpu < sizeof(long)*8-1 ; ++idxCpu){
+        if(CPU_ISSET(idxCpu, &mask)){
+            cpus |= (1 << idxCpu);
+        }
+    }
+    return cpus;
 #endif
     return -1;
 }
-- 
GitLab