OSDI Microbenchmarks
[akaros.git] / tests / msr_cycling_vcores.c
1 /* tests/msr_cycling_vcores.c
2  *
3  * This requests the max_vcores in the system, waits a bit, then gives them
4  * back, looping forever.  We can't give up all vcores, based on the current
5  * kernel, so we hold on to vcore0 to do the thinking. */
6
7 #include <ros/resource.h>
8 #include <parlib.h>
9 #include <rstdio.h>
10 #include <vcore.h>
11 #include <timing.h>
12 #include <mcs.h>
13
14 mcs_barrier_t b;
15 uint64_t begin = 0, end = 0;
16
17 int main(int argc, char** argv)
18 {
19
20         /* don't forget to enable notifs on vcore0.  if you don't, the kernel will
21          * restart your _S with notifs disabled, which is a path to confusion. */
22         struct preempt_data *vcpd = &__procdata.vcore_preempt_data[0];
23         vcpd->notif_enabled = TRUE;
24
25         mcs_barrier_init(&b, max_vcores());
26
27         vcore_request(max_vcores());
28         printf("not enough vcores, going to try it manually\n");
29         sys_resource_req(RES_CORES, max_vcores(), 1, REQ_SOFT);
30         printf("We're screwed!\n");
31
32         /* should never make it here */
33         return -1;
34 }
35
36 void vcore_entry(void)
37 {
38         uint32_t vcoreid = vcore_id();
39
40         if (vcoreid) {
41                 mcs_barrier_wait(&b, vcoreid);
42                 udelay(5000000);
43                 if (vcoreid == 1)
44                         printf("Proc %d's vcores are yielding\n", getpid());
45                 sys_yield(0);
46         } else {
47                 /* trip the barrier here, all future times are in the loop */
48                 mcs_barrier_wait(&b, vcoreid);
49                 while (1) {
50                         udelay(15000000);
51                         printf("Proc %d requesting its cores again\n", getpid());
52                         begin = read_tsc();
53                         sys_resource_req(RES_CORES, max_vcores(), 1, REQ_SOFT);
54                         mcs_barrier_wait(&b, vcoreid);
55                         end = read_tsc();
56                         printf("Took %llu usec (%llu nsec) to get my yielded cores back.\n",
57                                udiff(begin, end), ndiff(begin, end));
58                         printf("[T]:010:%llu:%llu\n",
59                                udiff(begin, end), ndiff(begin, end));
60                 }
61         }
62         printf("We're screwed!\n");
63         exit(-1);
64 }