Added jitter test

Author: HFTrader

libs/CMakeLists.txt

@@ -1,4 +1,4 @@
-set( LIBRARY_CPP_FILES Snapshot.cpp PerfGroup.cpp ) 
+set( LIBRARY_CPP_FILES Snapshot.cpp PerfGroup.cpp CpuUtils.cpp ) 
 set( LIBRARY_DEPENDENCIES pfm )
 if ( Armadillo_FOUND ) 
   list( APPEND LIBRARY_CPP_FILES Regression.cpp )
@@ -7,7 +7,7 @@ endif()
 add_library( tinyperfstats SHARED  ${LIBRARY_CPP_FILES} )
 target_link_libraries( tinyperfstats ${LIBRARY_DEPENDENCIES} )
 
-set( HEADER_LIST Allocators.h BitUtils.h DateUtils.h Histogram.h KahanSum.h MicroStats.h PerfCounter.h Snapshot.h StringUtils.h Ticker.h TimingHelpers.h Regression.h )
+set( HEADER_LIST Allocators.h BitUtils.h CpuUtils.h DateUtils.h Histogram.h KahanSum.h MicroStats.h PerfCounter.h Snapshot.h StringUtils.h Ticker.h TimingUtils.h Regression.h )
 foreach( HEADER ${HEADER_LIST} )
   file( REAL_PATH ${HEADER} ABSHEADER BASE_DIRECTORY ${CMAKE_CURRENT_SOURCE_DIR} )
   list( APPEND ALLHEADERS ${ABSHEADER} )

libs/CpuUtils.cpp

@@ -0,0 +1,84 @@
+#include "CpuUtils.h"
+#include <cstdio>
+
+std::vector<std::size_t> getThreadAffinity() {
+    std::vector<std::size_t> cores;
+    pthread_t self = pthread_self();
+    cpu_set_t cpuset;
+    CPU_ZERO(&cpuset);
+    ::pthread_getaffinity_np(self, sizeof(cpuset), &cpuset);
+    for (int j = 0; j < CPU_SETSIZE; j++) {
+        if (CPU_ISSET(j, &cpuset)) {
+            cores.push_back(j);
+        }
+    }
+    return cores;
+}
+
+bool setThreadAfinity(const std::vector<std::size_t>& cores) {
+    pthread_t self = pthread_self();
+    cpu_set_t mask;
+    CPU_ZERO(&mask);
+    for (std::size_t core : cores) {
+        CPU_SET(core, &mask);
+    }
+    int err = ::pthread_setaffinity_np(self, sizeof(mask), &mask);
+    return (err == 0);
+}
+
+bool setThreadAfinity(std::size_t core) {
+    pthread_t self = pthread_self();
+    cpu_set_t mask;
+    CPU_ZERO(&mask);
+    CPU_SET(core, &mask);
+    int err = ::pthread_setaffinity_np(self, sizeof(mask), &mask);
+    return (err == 0);
+}
+
+std::size_t getCurrentCore() {
+    return sched_getcpu();
+}
+
+std::size_t getNumberOfCores() {
+    return get_nprocs_conf();
+}
+
+const char* getPolicyName(int policy) {
+    switch (policy) {
+        case SCHED_RR: return "RoundRobin"; break;
+        case SCHED_BATCH: return "Batch"; break;
+        case SCHED_OTHER: return "Standard"; break;
+        case SCHED_IDLE: return "Idle"; break;
+        default: return "N/A"; break;
+    }
+}
+
+bool setPriority(int scheduler, int prio) {
+    pthread_t tid = pthread_self();
+    struct sched_param param;
+    int policy;
+    int res = pthread_getschedparam(tid, &policy, &param);
+    if (res != 0) {
+        perror("pthread_getschedparam");
+        return false;
+    }
+    param.sched_priority = prio;
+    res = pthread_setschedparam(tid, scheduler, &param);
+    if (res != 0) {
+        perror("pthread_setschedparam");
+        return false;
+    }
+    return true;
+}
+
+void setRealtimePriority(int prio) {
+    setPriority(SCHED_RR, prio);
+}
+
+void setIdlePriority(int prio) {
+    setPriority(SCHED_IDLE, prio);
+}
+
+void setStandardPriority(int prio) {
+    setPriority(SCHED_OTHER, prio);
+}

libs/CpuUtils.h

@@ -0,0 +1,21 @@
+#pragma once
+
+#include <pthread.h>
+#include <sys/sysinfo.h>
+#include <vector>
+
+std::vector<std::size_t> getThreadAffinity();
+
+bool setThreadAfinity(const std::vector<std::size_t>& cores);
+
+bool setThreadAfinity(std::size_t core);
+
+std::size_t getCurrentCore();
+
+std::size_t getNumberOfCores();
+
+const char* getPolicyName(int policy);
+bool setPriority(int scheduler, int prio);
+void setRealtimePriority(int prio);
+void setIdlePriority(int prio);
+void setStandardPriority(int prio);

libs/MicroStats.h

@@ -124,6 +124,16 @@ class MicroStats {
         return Range{base + offset, base + (base >> NDB) + offset - 1};
     }
 
+    //! Returns the number of events
+    uint64_t count() const {
+        return totalcount;
+    }
+
+    //! Returns the global average
+    double average() const {
+        return totalsum / totalcount;
+    }
+
 private:
     //! Initializes the histogram with the calculated ranges
     void init() {

libs/TimingHelpers.h libs/TimingUtils.hlibs/TimingHelpers.h renamed to libs/TimingUtils.h

@@ -1,13 +1,33 @@
 #pragma once
 
 #include <cstdint>
+#include <type_traits>
+#include <limits>
+#include <ctime>
+
+//! Returns the number of nanoseconds in UTC
+int64_t utcnow() {
+    struct timespec ts;
+    int res = ::clock_gettime(CLOCK_REALTIME, &ts);
+    if (res == 0) {
+        return int64_t(ts.tv_sec) * 1000000000L + int64_t(ts.tv_nsec);
+    }
+    return 0;
+}
 
 #if defined(__GNUC__) && defined(__x86_64__)
 
 #include <x86intrin.h>
 
 std::uint64_t tic() {
-    return __rdtsc();
+    return __builtin_ia32_rdtsc();
+}
+
+template <typename IntType,
+          typename RangeType = typename std::make_unsigned<IntType>::type>
+static inline RangeType forward_difference(const IntType& a, const IntType& b) {
+    if (a >= b) return a - b;
+    return (std::numeric_limits<IntType>::max() - a) + b;
 }
 
 static inline void mfence() {

tests/CMakeLists.txt

@@ -3,3 +3,4 @@ add_subdirectory( memorymap )
 add_subdirectory( cache )
 add_subdirectory( duffloop )
 add_subdirectory( microstats )
+add_subdirectory( jitter )

libs/Ticker.h tests/cache/Ticker.hlibs/Ticker.h renamed to tests/cache/Ticker.h

@@ -1,5 +1,5 @@
 
-#include "TimingHelpers.h"
+#include "TimingUtils.h"
 #include "Histogram.h"
 #include <x86intrin.h>
 

tests/cache/testCacheSize.cpp

@@ -2,7 +2,7 @@
 
 #include <cstddef>
 #include <cstdint>
-#include "TimingHelpers.h"
+#include "TimingUtils.h"
 #include "Histogram.h"
 
 // Defined in DuffLoopHelpers.cpp

tests/duffloop/testDuffLoop.cpp

@@ -0,0 +1,4 @@
+add_executable( jitter jitter.cpp )
+target_link_libraries( jitter tinyperfstats pthread )
+
+list( APPEND TARGETS jitter )