ec532e2c5c
The new feature, opt.percpu_arena, determines thread-arena association dynamically based CPU id. Three modes are supported: "percpu", "phycpu" and disabled. "percpu" uses the current core id (with help from sched_getcpu()) directly as the arena index, while "phycpu" will assign threads on the same physical CPU to the same arena. In other words, "percpu" means # of arenas == # of CPUs, while "phycpu" has # of arenas == 1/2 * (# of CPUs). Note that no runtime check on whether hyper threading is enabled is added yet. When enabled, threads will be migrated between arenas when a CPU change is detected. In the current design, to reduce overhead from reading CPU id, each arena tracks the thread accessed most recently. When a new thread comes in, we will read CPU id and update arena if necessary.
87 lines
1.8 KiB
C
87 lines
1.8 KiB
C
#include "test/jemalloc_test.h"
|
|
|
|
#define NTHREADS 10
|
|
|
|
void *
|
|
thd_start(void *arg) {
|
|
unsigned main_arena_ind = *(unsigned *)arg;
|
|
void *p;
|
|
unsigned arena_ind;
|
|
size_t size;
|
|
int err;
|
|
|
|
p = malloc(1);
|
|
assert_ptr_not_null(p, "Error in malloc()");
|
|
free(p);
|
|
|
|
size = sizeof(arena_ind);
|
|
if ((err = mallctl("thread.arena", (void *)&arena_ind, &size,
|
|
(void *)&main_arena_ind, sizeof(main_arena_ind)))) {
|
|
char buf[BUFERROR_BUF];
|
|
|
|
buferror(err, buf, sizeof(buf));
|
|
test_fail("Error in mallctl(): %s", buf);
|
|
}
|
|
|
|
size = sizeof(arena_ind);
|
|
if ((err = mallctl("thread.arena", (void *)&arena_ind, &size, NULL,
|
|
0))) {
|
|
char buf[BUFERROR_BUF];
|
|
|
|
buferror(err, buf, sizeof(buf));
|
|
test_fail("Error in mallctl(): %s", buf);
|
|
}
|
|
assert_u_eq(arena_ind, main_arena_ind,
|
|
"Arena index should be same as for main thread");
|
|
|
|
return NULL;
|
|
}
|
|
|
|
static void
|
|
mallctl_failure(int err) {
|
|
char buf[BUFERROR_BUF];
|
|
|
|
buferror(err, buf, sizeof(buf));
|
|
test_fail("Error in mallctl(): %s", buf);
|
|
}
|
|
|
|
TEST_BEGIN(test_thread_arena) {
|
|
void *p;
|
|
int err;
|
|
thd_t thds[NTHREADS];
|
|
unsigned i;
|
|
|
|
p = malloc(1);
|
|
assert_ptr_not_null(p, "Error in malloc()");
|
|
|
|
unsigned arena_ind, old_arena_ind;
|
|
size_t sz = sizeof(unsigned);
|
|
assert_d_eq(mallctl("arenas.create", (void *)&arena_ind, &sz, NULL, 0),
|
|
0, "Arena creation failure");
|
|
|
|
size_t size = sizeof(arena_ind);
|
|
if ((err = mallctl("thread.arena", (void *)&old_arena_ind, &size,
|
|
(void *)&arena_ind, sizeof(arena_ind))) != 0) {
|
|
mallctl_failure(err);
|
|
}
|
|
|
|
for (i = 0; i < NTHREADS; i++) {
|
|
thd_create(&thds[i], thd_start,
|
|
(void *)&arena_ind);
|
|
}
|
|
|
|
for (i = 0; i < NTHREADS; i++) {
|
|
intptr_t join_ret;
|
|
thd_join(thds[i], (void *)&join_ret);
|
|
assert_zd_eq(join_ret, 0, "Unexpected thread join error");
|
|
}
|
|
free(p);
|
|
}
|
|
TEST_END
|
|
|
|
int
|
|
main(void) {
|
|
return test(
|
|
test_thread_arena);
|
|
}
|