OSDI Microbenchmarks
[akaros.git] / tests / msr_get_cores.c
index 569d650..eeafb0e 100644 (file)
@@ -3,8 +3,13 @@
  * This measures the time it takes to request and receive the max_vcores() in
  * the system.  The clock starts before vcore_request(), which includes the time
  * it takes to allocate transition stacks and TLS.  The clock stops after
- * barriering in vcore_entry().  Alternatively, you can make vcore0 pop back out
- * and measure there (comment some things out in vcore entry()). */
+ * barriering in vcore_entry(), while vcore0 gets restarted and barriers.  You
+ * really want to restart vcore0's context so it releases the lock in
+ * vcore_request().
+ *
+ * This will measure both the hot and cold times, with the hot times not needing
+ * to have stacks mmaped and other things. */
+
 
 #include <parlib.h>
 #include <ros/mman.h>
@@ -27,6 +32,7 @@ uint64_t begin = 0, end = 0;
 int main(int argc, char** argv)
 {
        uint32_t vcoreid = vcore_id();
+       int retval = 0;
 
        mcs_barrier_init(&b, max_vcores());
 
@@ -51,13 +57,27 @@ int main(int argc, char** argv)
        vcpd->notif_enabled = TRUE;
 /* end: stuff userspace needs to do before switching to multi-mode */
 
-       /* should do this in the vcore entry */
        begin = read_tsc();
-       vcore_request(max_vcores());
+       retval = vcore_request(max_vcores());
+       if (retval)
+               printf("Fucked!\n");
        mcs_barrier_wait(&b, vcoreid);
        end = read_tsc();
 
-       printf("Took %llu usec (%llu nsec) to receive %d cores (restarting).\n",
+       printf("Took %llu usec (%llu nsec) to receive %d cores (cold).\n",
+              udiff(begin, end), ndiff(begin, end), max_vcores());
+       printf("[T]:001:%llu:%llu:%d:C.\n",
+              udiff(begin, end), ndiff(begin, end), max_vcores());
+       udelay(5000000);
+       begin = read_tsc();
+       retval = vcore_request(max_vcores() - 1);
+       if (retval)
+               printf("Fucked!\n");
+       mcs_barrier_wait(&b, vcoreid);
+       end = read_tsc();
+       printf("Took %llu usec (%llu nsec) to receive %d cores (warm).\n",
+              udiff(begin, end), ndiff(begin, end), max_vcores());
+       printf("[T]:001:%llu:%llu:%d:W.\n",
               udiff(begin, end), ndiff(begin, end), max_vcores());
 
        return 0;
@@ -65,21 +85,9 @@ int main(int argc, char** argv)
 
 void vcore_entry(void)
 {
-       uint32_t vcoreid = vcore_id(); // this will still be slow
-
-       /* try testing immediately.  remove from here to the while(1) if you want to
-        * count vcore0 restarting. */
-       mcs_barrier_wait(&b, vcoreid);
-       if (vcoreid == 0) {
-               end = read_tsc();
-               printf("Took %llu usec (%llu nsec) to receive %d cores (no restart).\n",
-                      udiff(begin, end), ndiff(begin, end), max_vcores());
-               exit(0);
-       }
-       while(1);
+       uint32_t vcoreid = vcore_id();
 
 /* begin: stuff userspace needs to do to handle notifications */
-
        struct vcore *vc = &__procinfo.vcoremap[vcoreid];
        struct preempt_data *vcpd;
        vcpd = &__procdata.vcore_preempt_data[vcoreid];
@@ -94,15 +102,16 @@ void vcore_entry(void)
                pop_ros_tf(&vcpd->notif_tf, vcoreid);
                panic("should never see me!");
        }       
-       /* unmask notifications once you can let go of the notif_tf and it is okay
-        * to clobber the transition stack.
-        * Check Documentation/processes.txt: 4.2.4.  In real code, you should be
-        * popping the tf of whatever user process you want (get off the x-stack) */
-       vcpd->notif_enabled = TRUE;
-       
 /* end: stuff userspace needs to do to handle notifications */
+
+       /* all other vcores are down here */
        mcs_barrier_wait(&b, vcoreid);
-       while(1);
 
+       udelay(1000000);
+       if (vcoreid == 1)
+               printf("Proc %d's vcores are yielding\n", getpid());
+       sys_yield(0);
+
+       while(1);
 }