|
Packit |
6c4009 |
/* Benchmark malloc and free functions.
|
|
Packit |
6c4009 |
Copyright (C) 2013-2018 Free Software Foundation, Inc.
|
|
Packit |
6c4009 |
This file is part of the GNU C Library.
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
The GNU C Library is free software; you can redistribute it and/or
|
|
Packit |
6c4009 |
modify it under the terms of the GNU Lesser General Public
|
|
Packit |
6c4009 |
License as published by the Free Software Foundation; either
|
|
Packit |
6c4009 |
version 2.1 of the License, or (at your option) any later version.
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
The GNU C Library is distributed in the hope that it will be useful,
|
|
Packit |
6c4009 |
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
Packit |
6c4009 |
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
Packit |
6c4009 |
Lesser General Public License for more details.
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
You should have received a copy of the GNU Lesser General Public
|
|
Packit |
6c4009 |
License along with the GNU C Library; if not, see
|
|
Packit |
6c4009 |
<http://www.gnu.org/licenses/>. */
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
#include <errno.h>
|
|
Packit |
6c4009 |
#include <math.h>
|
|
Packit |
6c4009 |
#include <pthread.h>
|
|
Packit |
6c4009 |
#include <signal.h>
|
|
Packit |
6c4009 |
#include <stdio.h>
|
|
Packit |
6c4009 |
#include <stdlib.h>
|
|
Packit |
6c4009 |
#include <string.h>
|
|
Packit |
6c4009 |
#include <sys/time.h>
|
|
Packit |
6c4009 |
#include <sys/resource.h>
|
|
Packit |
6c4009 |
#include <unistd.h>
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
#include "bench-timing.h"
|
|
Packit |
6c4009 |
#include "json-lib.h"
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
/* Benchmark duration in seconds. */
|
|
Packit |
6c4009 |
#define BENCHMARK_DURATION 60
|
|
Packit |
6c4009 |
#define RAND_SEED 88
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
#ifndef NUM_THREADS
|
|
Packit |
6c4009 |
# define NUM_THREADS 1
|
|
Packit |
6c4009 |
#endif
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
/* Maximum memory that can be allocated at any one time is:
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
NUM_THREADS * WORKING_SET_SIZE * MAX_ALLOCATION_SIZE
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
However due to the distribution of the random block sizes
|
|
Packit |
6c4009 |
the typical amount allocated will be much smaller. */
|
|
Packit |
6c4009 |
#define WORKING_SET_SIZE 1024
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
#define MIN_ALLOCATION_SIZE 4
|
|
Packit |
6c4009 |
#define MAX_ALLOCATION_SIZE 32768
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
/* Get a random block size with an inverse square distribution. */
|
|
Packit |
6c4009 |
static unsigned int
|
|
Packit |
6c4009 |
get_block_size (unsigned int rand_data)
|
|
Packit |
6c4009 |
{
|
|
Packit |
6c4009 |
/* Inverse square. */
|
|
Packit |
6c4009 |
const float exponent = -2;
|
|
Packit |
6c4009 |
/* Minimum value of distribution. */
|
|
Packit |
6c4009 |
const float dist_min = MIN_ALLOCATION_SIZE;
|
|
Packit |
6c4009 |
/* Maximum value of distribution. */
|
|
Packit |
6c4009 |
const float dist_max = MAX_ALLOCATION_SIZE;
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
float min_pow = powf (dist_min, exponent + 1);
|
|
Packit |
6c4009 |
float max_pow = powf (dist_max, exponent + 1);
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
float r = (float) rand_data / RAND_MAX;
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
return (unsigned int) powf ((max_pow - min_pow) * r + min_pow,
|
|
Packit |
6c4009 |
1 / (exponent + 1));
|
|
Packit |
6c4009 |
}
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
#define NUM_BLOCK_SIZES 8000
|
|
Packit |
6c4009 |
#define NUM_OFFSETS ((WORKING_SET_SIZE) * 4)
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
static unsigned int random_block_sizes[NUM_BLOCK_SIZES];
|
|
Packit |
6c4009 |
static unsigned int random_offsets[NUM_OFFSETS];
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
static void
|
|
Packit |
6c4009 |
init_random_values (void)
|
|
Packit |
6c4009 |
{
|
|
Packit |
6c4009 |
for (size_t i = 0; i < NUM_BLOCK_SIZES; i++)
|
|
Packit |
6c4009 |
random_block_sizes[i] = get_block_size (rand ());
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
for (size_t i = 0; i < NUM_OFFSETS; i++)
|
|
Packit |
6c4009 |
random_offsets[i] = rand () % WORKING_SET_SIZE;
|
|
Packit |
6c4009 |
}
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
static unsigned int
|
|
Packit |
6c4009 |
get_random_block_size (unsigned int *state)
|
|
Packit |
6c4009 |
{
|
|
Packit |
6c4009 |
unsigned int idx = *state;
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
if (idx >= NUM_BLOCK_SIZES - 1)
|
|
Packit |
6c4009 |
idx = 0;
|
|
Packit |
6c4009 |
else
|
|
Packit |
6c4009 |
idx++;
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
*state = idx;
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
return random_block_sizes[idx];
|
|
Packit |
6c4009 |
}
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
static unsigned int
|
|
Packit |
6c4009 |
get_random_offset (unsigned int *state)
|
|
Packit |
6c4009 |
{
|
|
Packit |
6c4009 |
unsigned int idx = *state;
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
if (idx >= NUM_OFFSETS - 1)
|
|
Packit |
6c4009 |
idx = 0;
|
|
Packit |
6c4009 |
else
|
|
Packit |
6c4009 |
idx++;
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
*state = idx;
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
return random_offsets[idx];
|
|
Packit |
6c4009 |
}
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
static volatile bool timeout;
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
static void
|
|
Packit |
6c4009 |
alarm_handler (int signum)
|
|
Packit |
6c4009 |
{
|
|
Packit |
6c4009 |
timeout = true;
|
|
Packit |
6c4009 |
}
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
/* Allocate and free blocks in a random order. */
|
|
Packit |
6c4009 |
static size_t
|
|
Packit |
6c4009 |
malloc_benchmark_loop (void **ptr_arr)
|
|
Packit |
6c4009 |
{
|
|
Packit |
6c4009 |
unsigned int offset_state = 0, block_state = 0;
|
|
Packit |
6c4009 |
size_t iters = 0;
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
while (!timeout)
|
|
Packit |
6c4009 |
{
|
|
Packit |
6c4009 |
unsigned int next_idx = get_random_offset (&offset_state);
|
|
Packit |
6c4009 |
unsigned int next_block = get_random_block_size (&block_state);
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
free (ptr_arr[next_idx]);
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
ptr_arr[next_idx] = malloc (next_block);
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
iters++;
|
|
Packit |
6c4009 |
}
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
return iters;
|
|
Packit |
6c4009 |
}
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
struct thread_args
|
|
Packit |
6c4009 |
{
|
|
Packit |
6c4009 |
size_t iters;
|
|
Packit |
6c4009 |
void **working_set;
|
|
Packit |
6c4009 |
timing_t elapsed;
|
|
Packit |
6c4009 |
};
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
static void *
|
|
Packit |
6c4009 |
benchmark_thread (void *arg)
|
|
Packit |
6c4009 |
{
|
|
Packit |
6c4009 |
struct thread_args *args = (struct thread_args *) arg;
|
|
Packit |
6c4009 |
size_t iters;
|
|
Packit |
6c4009 |
void *thread_set = args->working_set;
|
|
Packit |
6c4009 |
timing_t start, stop;
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
TIMING_NOW (start);
|
|
Packit |
6c4009 |
iters = malloc_benchmark_loop (thread_set);
|
|
Packit |
6c4009 |
TIMING_NOW (stop);
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
TIMING_DIFF (args->elapsed, start, stop);
|
|
Packit |
6c4009 |
args->iters = iters;
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
return NULL;
|
|
Packit |
6c4009 |
}
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
static timing_t
|
|
Packit |
6c4009 |
do_benchmark (size_t num_threads, size_t *iters)
|
|
Packit |
6c4009 |
{
|
|
Packit |
6c4009 |
timing_t elapsed = 0;
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
if (num_threads == 1)
|
|
Packit |
6c4009 |
{
|
|
Packit |
6c4009 |
timing_t start, stop;
|
|
Packit |
6c4009 |
void *working_set[WORKING_SET_SIZE];
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
memset (working_set, 0, sizeof (working_set));
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
TIMING_NOW (start);
|
|
Packit |
6c4009 |
*iters = malloc_benchmark_loop (working_set);
|
|
Packit |
6c4009 |
TIMING_NOW (stop);
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
TIMING_DIFF (elapsed, start, stop);
|
|
Packit |
6c4009 |
}
|
|
Packit |
6c4009 |
else
|
|
Packit |
6c4009 |
{
|
|
Packit |
6c4009 |
struct thread_args args[num_threads];
|
|
Packit |
6c4009 |
void *working_set[num_threads][WORKING_SET_SIZE];
|
|
Packit |
6c4009 |
pthread_t threads[num_threads];
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
memset (working_set, 0, sizeof (working_set));
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
*iters = 0;
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
for (size_t i = 0; i < num_threads; i++)
|
|
Packit |
6c4009 |
{
|
|
Packit |
6c4009 |
args[i].working_set = working_set[i];
|
|
Packit |
6c4009 |
pthread_create(&threads[i], NULL, benchmark_thread, &args[i]);
|
|
Packit |
6c4009 |
}
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
for (size_t i = 0; i < num_threads; i++)
|
|
Packit |
6c4009 |
{
|
|
Packit |
6c4009 |
pthread_join(threads[i], NULL);
|
|
Packit |
6c4009 |
TIMING_ACCUM (elapsed, args[i].elapsed);
|
|
Packit |
6c4009 |
*iters += args[i].iters;
|
|
Packit |
6c4009 |
}
|
|
Packit |
6c4009 |
}
|
|
Packit |
6c4009 |
return elapsed;
|
|
Packit |
6c4009 |
}
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
static void usage(const char *name)
|
|
Packit |
6c4009 |
{
|
|
Packit |
6c4009 |
fprintf (stderr, "%s: <num_threads>\n", name);
|
|
Packit |
6c4009 |
exit (1);
|
|
Packit |
6c4009 |
}
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
int
|
|
Packit |
6c4009 |
main (int argc, char **argv)
|
|
Packit |
6c4009 |
{
|
|
Packit |
6c4009 |
timing_t cur;
|
|
Packit |
6c4009 |
size_t iters = 0, num_threads = 1;
|
|
Packit |
6c4009 |
unsigned long res;
|
|
Packit |
6c4009 |
json_ctx_t json_ctx;
|
|
Packit |
6c4009 |
double d_total_s, d_total_i;
|
|
Packit |
6c4009 |
struct sigaction act;
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
if (argc == 1)
|
|
Packit |
6c4009 |
num_threads = 1;
|
|
Packit |
6c4009 |
else if (argc == 2)
|
|
Packit |
6c4009 |
{
|
|
Packit |
6c4009 |
long ret;
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
errno = 0;
|
|
Packit |
6c4009 |
ret = strtol(argv[1], NULL, 10);
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
if (errno || ret == 0)
|
|
Packit |
6c4009 |
usage(argv[0]);
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
num_threads = ret;
|
|
Packit |
6c4009 |
}
|
|
Packit |
6c4009 |
else
|
|
Packit |
6c4009 |
usage(argv[0]);
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
init_random_values ();
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
json_init (&json_ctx, 0, stdout);
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
json_document_begin (&json_ctx);
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
json_attr_string (&json_ctx, "timing_type", TIMING_TYPE);
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
json_attr_object_begin (&json_ctx, "functions");
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
json_attr_object_begin (&json_ctx, "malloc");
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
json_attr_object_begin (&json_ctx, "");
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
TIMING_INIT (res);
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
(void) res;
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
memset (&act, 0, sizeof (act));
|
|
Packit |
6c4009 |
act.sa_handler = &alarm_handler;
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
sigaction (SIGALRM, &act, NULL);
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
alarm (BENCHMARK_DURATION);
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
cur = do_benchmark (num_threads, &iters);
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
struct rusage usage;
|
|
Packit |
6c4009 |
getrusage(RUSAGE_SELF, &usage);
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
d_total_s = cur;
|
|
Packit |
6c4009 |
d_total_i = iters;
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
json_attr_double (&json_ctx, "duration", d_total_s);
|
|
Packit |
6c4009 |
json_attr_double (&json_ctx, "iterations", d_total_i);
|
|
Packit |
6c4009 |
json_attr_double (&json_ctx, "time_per_iteration", d_total_s / d_total_i);
|
|
Packit |
6c4009 |
json_attr_double (&json_ctx, "max_rss", usage.ru_maxrss);
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
json_attr_double (&json_ctx, "threads", num_threads);
|
|
Packit |
6c4009 |
json_attr_double (&json_ctx, "min_size", MIN_ALLOCATION_SIZE);
|
|
Packit |
6c4009 |
json_attr_double (&json_ctx, "max_size", MAX_ALLOCATION_SIZE);
|
|
Packit |
6c4009 |
json_attr_double (&json_ctx, "random_seed", RAND_SEED);
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
json_attr_object_end (&json_ctx);
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
json_attr_object_end (&json_ctx);
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
json_attr_object_end (&json_ctx);
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
json_document_end (&json_ctx);
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
return 0;
|
|
Packit |
6c4009 |
}
|