trunk/markers-test/test-mark-speed-opt.c

   1 /* test-mark.c
   2  *
   3  */
   4
   5 #include <linux/module.h>
   6 #include <linux/proc_fs.h>
   7 #include <linux/sched.h>
   8 #include <linux/timex.h>
   9 #include <linux/marker.h>
  10 #include <asm/ptrace.h>
  11 #include <asm/system.h>
  12
  13 static void pmc_flush_cache(void)
  14   {
  15     register int i;
  16     /* write back and invalidate cache (a serializing instruction) */
  17
  18     __asm__ __volatile__ ( "wbinvd" : : : "memory" );
  19
  20     /* The wbinvd instruction does not wait for the external caches
  21      * to be flushed, but only requests that it be done.  The loop
  22      * is to be sure that enough time has elapsed, but the compiler
  23      * might simplify or even remove it.  The loop bound is for a
  24      * 512 KB L2 cache.  On a Pentium Pro/II/III, the loop uses
  25      * 2 cycles per iteration.
  26      *
  27      * Does wbinvd also cause the TLB to be flushed?
  28      * A comment in mtrr.c suggests that it does.
  29      */
  30     for (i = 0; i < 512*1024; i++) {
  31       cpu_relax();
  32     }
  33   }
  34
  35
  36 static void noinline test2(const struct marker *mdata,
  37         void *call_private, ...)
  38 {
  39         /* not called */
  40         printk("blah\n");
  41 }
  42
  43 /*
  44  * Generic marker flavor always available.
  45  * Note : the empty asm volatile with read constraint is used here instead of a
  46  * "used" attribute to fix a gcc 4.1.x bug.
  47  * Make sure the alignment of the structure in the __markers section will
  48  * not add unwanted padding between the beginning of the section and the
  49  * structure. Force alignment to the same alignment as the section start.
  50  */
  51 #define __my_trace_mark(generic, name, call_private, format, args...)   \
  52         do {                                                            \
  53                 static const char __mstrtab_##name[]                    \
  54                 __attribute__((section("__markers_strings")))           \
  55                 = #name "\0" format;                                    \
  56                 static struct marker __mark_##name                      \
  57                 __attribute__((section("__markers"), aligned(8))) =     \
  58                 { __mstrtab_##name, &__mstrtab_##name[sizeof(#name)],   \
  59                 0, 0, marker_probe_cb,                                  \
  60                 { __mark_empty_function, NULL}, NULL };                 \
  61                 __mark_check_format(format, ## args);                   \
  62                 if (!generic) {                                         \
  63                         if (unlikely(imv_read(__mark_##name.state)))    \
  64                                 test2                   \
  65                                         (&__mark_##name, call_private,  \
  66                                         ## args);               \
  67                 } else {                                                \
  68                         if (unlikely(_imv_read(__mark_##name.state)))   \
  69                                 test2                   \
  70                                         (&__mark_##name, call_private,  \
  71                                         ## args);               \
  72                 }                                                       \
  73         } while (0)
  74
  75 struct proc_dir_entry *pentry = NULL;
  76
  77 static inline void test(unsigned long arg, unsigned long arg2)
  78 {
  79         volatile int temp[5];
  80 #ifdef CACHEFLUSH
  81         clflush(&current->pid);
  82         //pmc_flush_cache();
  83 #endif
  84         temp[2] = (temp[0] + 60) << 10;
  85         temp[3] = (temp[2] + 60) << 10;
  86         temp[4] = (temp[3] + 60) << 10;
  87         temp[0] = (temp[4] + 60) << 10;
  88         //asm volatile ("");
  89         //__my_trace_mark(1, kernel_debug_test, NULL, "%d %d %ld %ld", 2, current->pid, arg, arg2);
  90         barrier();
  91         __my_trace_mark(0, kernel_debug_test, NULL, "%d %d %ld %ld", 2, current->pid, arg, arg2);
  92         barrier();
  93 }
  94
  95 static int my_open(struct inode *inode, struct file *file)
  96 {
  97         unsigned int i;
  98         cycles_t cycles1, cycles2;
  99         unsigned long flags;
 100
 101         local_irq_save(flags);
 102 #ifdef CACHEFLUSH
 103         //pmc_flush_cache();    /* initial write back, without cycle count */
 104         //msleep(20);   /* wait for L2 flush */
 105 #endif
 106         rdtsc_barrier();
 107         cycles1 = get_cycles();
 108         rdtsc_barrier();
 109         for(i=0; i<2000; i++) {
 110                 test(i, i);
 111                 test(i, i);
 112                 test(i, i);
 113                 test(i, i);
 114                 test(i, i);
 115                 test(i, i);
 116                 test(i, i);
 117                 test(i, i);
 118                 test(i, i);
 119                 test(i, i);
 120         }
 121         rdtsc_barrier();
 122         cycles2 = get_cycles();
 123         rdtsc_barrier();
 124         local_irq_restore(flags);
 125         printk("cycles : %llu\n", cycles2-cycles1);
 126         return -EPERM;
 127 }
 128
 129
 130 static struct file_operations my_operations = {
 131         .open = my_open,
 132 };
 133
 134 int init_module(void)
 135 {
 136         pentry = create_proc_entry("testmark", 0444, NULL);
 137         if (pentry)
 138                 pentry->proc_fops = &my_operations;
 139
 140         return 0;
 141 }
 142
 143 void cleanup_module(void)
 144 {
 145         remove_proc_entry("testmark", NULL);
 146 }
 147
 148 MODULE_LICENSE("GPL");
 149 MODULE_AUTHOR("Mathieu Desnoyers");
 150 MODULE_DESCRIPTION("Marker Test");
 151 MODULE_VERSION("1.0");
 152