uhd: Add thread affinity utility functions

1 files changed, 43 insertions, 0 deletions

diff --git a/host/lib/utils/CMakeLists.txt b/host/lib/utils/CMakeLists.txt
index 74b292fbb..5c4478909 100644
--- a/host/lib/utils/CMakeLists.txt
+++ b/host/lib/utils/CMakeLists.txt
@@ -55,6 +55,10 @@ else()
     set(THREAD_PRIO_DEFS HAVE_THREAD_PRIO_DUMMY)
 endif()
 
+########################################################################
+# Setup defines for thread naming
+########################################################################
+
 set(CMAKE_REQUIRED_LIBRARIES "pthread")
 
 CHECK_CXX_SOURCE_COMPILES("
@@ -82,6 +86,45 @@ else()
     list(APPEND THREAD_PRIO_DEFS HAVE_THREAD_SETNAME_DUMMY)
 endif()
 
+########################################################################
+# Setup defines for thread affinitizing
+########################################################################
+
+CHECK_CXX_SOURCE_COMPILES("
+    #include <pthread.h>
+    int main(){
+        pthread_t pt;
+        cpu_set_t cs;
+        pthread_setaffinity_np(pt, sizeof(cpu_set_t), &cs);
+        return 0;
+    }
+    " HAVE_PTHREAD_SETAFFINITY_NP
+)
+
+if(CYGWIN)
+    set(HAVE_PTHREAD_SETAFFINITY_NP False)
+endif(CYGWIN)
+
+CHECK_CXX_SOURCE_COMPILES("
+    #include <windows.h>
+    int main(){
+        SetThreadAffinityMask(GetCurrentThread(), 0);
+        return 0;
+    }
+    " HAVE_WIN_SETTHREADAFFINITYMASK
+)
+if(HAVE_PTHREAD_SETAFFINITY_NP)
+    message(STATUS "  Setting thread affinity is supported through pthread_setaffinity_np.")
+    list(APPEND THREAD_PRIO_DEFS HAVE_PTHREAD_SETAFFINITY_NP)
+    LIBUHD_APPEND_LIBS(pthread)
+elseif(HAVE_WIN_SETTHREADPRIORITY)
+    message(STATUS "  Setting thread affinity is supported through windows SetThreadAffinityMask.")
+    list(APPEND THREAD_PRIO_DEFS HAVE_WIN_SETTHREADAFFINITYMASK)
+else()
+    message(STATUS "  Setting thread affinity is not supported.")
+    list(APPEND THREAD_PRIO_DEFS HAVE_THREAD_SETAFFINITY_DUMMY)
+endif()
+
 
 set_source_files_properties(
     ${CMAKE_CURRENT_SOURCE_DIR}/thread.cpp