void *yield_thread(void* arg)
{
/* Wait til all threads are created */
while (!ready)
cpu_relax();
for (int i = 0; i < nr_yield_loops; i++) {
printf_safe("[A] pthread %d %p on vcore %d, itr: %d\n", pthread_self()->id,
pthread_self(), vcore_id(), i);
/* Fakes some work by spinning a bit. Amount varies per uth/vcore,
* scaled by fake_work */
if (amt_fake_work)
udelay(amt_fake_work * (pthread_self()->id * (vcore_id() + 1)));
pthread_yield();
printf_safe("[A] pthread %p returned from yield on vcore %d, itr: %d\n",
pthread_self(), vcore_id(), i);
}
return (void*)(pthread_self());
}
void *yield_thread(void* arg)
{
/* Wait til all threads are created */
pthread_barrier_wait(&barrier);
for (int i = 0; i < nr_yield_loops; i++) {
printf_safe("[A] pthread %d %p on vcore %d, itr: %d\n",
pthread_id(), pthread_self(), vcore_id(), i);
/* Fakes some work by spinning a bit. Amount varies per
* uth/vcore, scaled by fake_work */
if (amt_fake_work)
udelay(amt_fake_work * (pthread_id() * (vcore_id() +
2)));
pthread_yield();
printf_safe("[A] pthread %p returned from yield on vcore %d, itr: %d\n",
pthread_self(), vcore_id(), i);
}
return (void*)(pthread_self());
}
void *block_thread(void* arg)
{
assert(!in_vcore_context());
for (int i = 0; i < NUM_TEST_LOOPS; i++) {
printf_safe("[A] pthread %d on vcore %d\n", pthread_self()->id, vcore_id());
sys_block(5000 + pthread_self()->id);
}
return (void*)(long)pthread_self()->id;
}
int main(int argc, char** argv)
{
struct timeval tv = {0};
if (gettimeofday(&tv, 0))
perror("Time error...");
printf("Start time: %dsec %dusec\n", tv.tv_sec, tv.tv_usec);
for (int i = 0; i < NUM_TEST_THREADS; i++) {
printf_safe("[A] About to create thread %d\n", i);
pthread_create(&my_threads[i], NULL, &block_thread, NULL);
}
for (int i = 0; i < NUM_TEST_THREADS; i++) {
printf_safe("[A] About to join on thread %d\n", i);
pthread_join(my_threads[i], &my_retvals[i]);
printf_safe("[A] Successfully joined on thread %d (retval: %p)\n", i,
my_retvals[i]);
}
if (gettimeofday(&tv, 0))
perror("Time error...");
printf("End time : %dsec %dusec\n", tv.tv_sec, tv.tv_usec);
printf("All done, exiting cleanishly\n");
}
int main(int argc, char** argv)
{
struct timeval start_tv = {0};
struct timeval end_tv = {0};
long usec_diff;
long nr_ctx_switches;
if (argc > 1)
nr_yield_threads = strtol(argv[1], 0, 10);
if (argc > 2)
nr_yield_loops = strtol(argv[2], 0, 10);
if (argc > 3)
nr_vcores = strtol(argv[3], 0, 10);
if (argc > 4)
amt_fake_work = strtol(argv[4], 0, 10);
nr_yield_threads = MIN(nr_yield_threads, MAX_NR_TEST_THREADS);
printf("Making %d threads of %d loops each, on %d vcore(s), %d work\n",
nr_yield_threads, nr_yield_loops, nr_vcores, amt_fake_work);
/* OS dependent prep work */
if (nr_vcores) {
/* Only do the vcore trickery if requested */
pthread_can_vcore_request(FALSE); /* 2LS won't manage vcores */
pthread_need_tls(FALSE);
pthread_lib_init(); /* gives us one vcore */
vcore_request(nr_vcores - 1); /* ghetto incremental interface */
for (int i = 0; i < nr_vcores; i++) {
printd("Vcore %d mapped to pcore %d\n", i,
__procinfo.vcoremap[i].pcoreid);
}
}
/* create and join on yield */
for (int i = 0; i < nr_yield_threads; i++) {
printf_safe("[A] About to create thread %d\n", i);
if (pthread_create(&my_threads[i], NULL, &yield_thread, NULL))
perror("pth_create failed");
}
if (gettimeofday(&start_tv, 0))
perror("Start time error...");
ready = TRUE; /* signal to any spinning uthreads to start */
for (int i = 0; i < nr_yield_threads; i++) {
printf_safe("[A] About to join on thread %d(%p)\n", i, my_threads[i]);
pthread_join(my_threads[i], &my_retvals[i]);
printf_safe("[A] Successfully joined on thread %d (retval: %p)\n", i,
my_retvals[i]);
}
if (gettimeofday(&end_tv, 0))
perror("End time error...");
nr_ctx_switches = nr_yield_threads * nr_yield_loops;
usec_diff = (end_tv.tv_sec - start_tv.tv_sec) * 1000000 +
(end_tv.tv_usec - start_tv.tv_usec);
printf("Done: %d uthreads, %d loops, %d vcores, %d work\n",
nr_yield_threads, nr_yield_loops, nr_vcores, amt_fake_work);
printf("Nr context switches: %d\n", nr_ctx_switches);
printf("Time to run: %d usec\n", usec_diff);
if (nr_vcores == 1)
printf("Context switch latency: %d nsec\n",
(int)(1000LL*usec_diff / nr_ctx_switches));
printf("Context switches / sec: %d\n\n",
(int)(1000000LL*nr_ctx_switches / usec_diff));
}
int main(int argc, char** argv)
{
struct timeval start_tv = {0};
struct timeval end_tv = {0};
long usec_diff;
long nr_ctx_switches;
if (argc > 1)
nr_yield_threads = strtol(argv[1], 0, 10);
if (argc > 2)
nr_yield_loops = strtol(argv[2], 0, 10);
if (argc > 3)
nr_vcores = strtol(argv[3], 0, 10);
if (argc > 4)
amt_fake_work = strtol(argv[4], 0, 10);
nr_yield_threads = MIN(nr_yield_threads, MAX_NR_TEST_THREADS);
printf("Making %d threads of %d loops each, on %d vcore(s), %d work\n",
nr_yield_threads, nr_yield_loops, nr_vcores, amt_fake_work);
/* OS dependent prep work */
#ifdef __ros__
if (nr_vcores) {
/* Only do the vcore trickery if requested */
parlib_never_yield = TRUE;
pthread_need_tls(FALSE);
pthread_mcp_init(); /* gives us one vcore */
vcore_request_total(nr_vcores);
parlib_never_vc_request = TRUE;
for (int i = 0; i < nr_vcores; i++) {
printf_safe("Vcore %d mapped to pcore %d\n", i,
__procinfo.vcoremap[i].pcoreid);
}
}
struct uth_join_request *join_reqs;
join_reqs = malloc(nr_yield_threads * sizeof(struct uth_join_request));
for (int i = 0; i < nr_yield_threads; i++)
join_reqs[i].retval_loc = &my_retvals[i];
assert(join_reqs);
#endif /* __ros__ */
pthread_barrier_init(&barrier, NULL, nr_yield_threads);
/* create and join on yield */
for (int i = 0; i < nr_yield_threads; i++) {
printf_safe("[A] About to create thread %d\n", i);
if (pthread_create(&my_threads[i], NULL, &yield_thread, NULL))
perror("pth_create failed");
}
if (gettimeofday(&start_tv, 0))
perror("Start time error...");
/* Akaros supports parallel join */
#ifdef __ros__
for (int i = 0; i < nr_yield_threads; i++)
join_reqs[i].uth = (struct uthread*)my_threads[i];
uthread_join_arr(join_reqs, nr_yield_threads);
#else
for (int i = 0; i < nr_yield_threads; i++) {
printf_safe("[A] About to join on thread %d(%p)\n",
i, my_threads[i]);
pthread_join(my_threads[i], &my_retvals[i]);
printf_safe("[A] Successful join on thread %d (retval: %p)\n",
i, my_retvals[i]);
}
#endif
if (gettimeofday(&end_tv, 0))
perror("End time error...");
nr_ctx_switches = nr_yield_threads * nr_yield_loops;
usec_diff = (end_tv.tv_sec - start_tv.tv_sec) * 1000000 +
(end_tv.tv_usec - start_tv.tv_usec);
printf("Done: %d uthreads, %d loops, %d vcores, %d work\n",
nr_yield_threads, nr_yield_loops, nr_vcores, amt_fake_work);
printf("Nr context switches: %ld\n", nr_ctx_switches);
printf("Time to run: %ld usec\n", usec_diff);
if (nr_vcores == 1)
printf("Context switch latency: %d nsec\n",
(int)(1000LL*usec_diff / nr_ctx_switches));
printf("Context switches / sec: %d\n\n",
(int)(1000000LL*nr_ctx_switches / usec_diff));
}
