diff --git a/bench/bench-mlib.c b/bench/bench-mlib.c index f17b9363..6bfbc9b2 100644 --- a/bench/bench-mlib.c +++ b/bench/bench-mlib.c @@ -347,6 +347,89 @@ static void test_buffer(size_t n) buffer_uint_clear(g_buff); } +/********************************************************************************************/ + +QUEUE_MPMC_DEF(queue_uint, unsigned int, BUFFER_QUEUE) +queue_uint_t g_buff2; + +QUEUE_MPMC_DEF(queue_ull, unsigned long long, BUFFER_QUEUE) +queue_ull_t g_final2; + +static void final2(void *arg) +{ + size_t *p_n = arg; + size_t n = *p_n; + unsigned long long j, s = 0; + for(int i = 0; i < n;i++) { + while (!queue_ull_pop(&j, g_final2)); + s += j; + } + g_result = s; +} + +static void conso2(void *arg) +{ + unsigned int j; + size_t *p_n = arg; + size_t n = *p_n; + unsigned long long s = 0; + for(int i = 0; i < n;i++) { + while (!queue_uint_pop(&j, g_buff2)); + s += j; + } + while (!queue_ull_push(g_final2, s)); +} + +static void prod2(void *arg) +{ + size_t *p_n = arg; + size_t n = *p_n; + size_t r = n; + for(unsigned int i = 0; i < n;i++) { + while (!queue_uint_push(g_buff2, r )); + r = r * 31421U + 6927U; + } +} + +static void test_queue(size_t n) +{ + const int cpu_count = get_cpu_count(); + const int prod_count = cpu_count/2; + const int conso_count = cpu_count - prod_count; + if (cpu_count < 2) { + fprintf(stderr, "WARNING: Can not measure Queue performance.\n"); + return; + } + // Init + queue_uint_init(g_buff2, 64*cpu_count); + queue_ull_init (g_final2, 64*cpu_count); + + // Create thread + m_thread_t idx_p[prod_count]; + m_thread_t idx_c[conso_count]; + m_thread_t idx_final; + for(int i = 0; i < prod_count; i++) { + m_thread_create (idx_p[i], prod2, &n); + } + for(int i = 0; i < conso_count; i++) { + m_thread_create (idx_c[i], conso2, &n); + } + size_t n2 = conso_count; + m_thread_create(idx_final, final2, &n2); + + // Wait for jobs to be done. + for(int i = 0; i < prod_count; i++) { + m_thread_join(idx_p[i]); + } + for(int i = 0; i < conso_count; i++) { + m_thread_join(idx_c[i]); + } + m_thread_join(idx_final); + + // Clear & quit + queue_ull_clear(g_final2); + queue_uint_clear(g_buff2); +} /********************************************************************************************/ @@ -404,6 +487,8 @@ int main(int argc, const char *argv[]) test_function("Sort time", 10000000, test_sort); if (n == 60) test_function("Buffer time", 1000000, test_buffer); + if (n == 61) + test_function("Queue MPMC time", 1000000, test_queue); if (n == 70) { n = (argc > 2) ? atoi(argv[2]) : 100000000; test_hash_prepare(n);