initial toy version of timing tools

.gitignore

@@ -0,0 +1,5 @@
+*.o
+
+
+*.asm
+*.so

README.md

@@ -0,0 +1,2 @@
+=C Arbitrary Timing System=
+Time stuff, really accurately, really easily.

cats.py

@@ -0,0 +1,110 @@
+import argparse
+import subprocess
+
+HOST_C_FILE = "timer.c"
+
+def print_info(s):
+    print "[CATS] "+s
+
+def print_unimp(fn):
+    print_info(fn+" is unimplemented.")
+
+def replace_file(fin,fout,reps):
+    fi = open(fin,'r')
+    fo = open(fout,'w')
+
+    for l in fi:
+        for (t,r) in reps:
+            if t in l:
+                l = l.replace(t,r)
+        fo.write(l)
+    fi.close()
+    fo.close()
+
+def gcc_build(args):
+    gccstring = "gcc -O -fforce-addr -std=c99 -o Cats_out "+args.tmpfile
+    proc = subprocess.Popen(gccstring.split(' '))
+
+def c_snippet(args):
+
+    ret_s = args.typestring.split('(')[0]
+    types_s = args.typestring.split('(')[1].split(')')[0]
+    argnames = map(chr, range(0x61, 0x61+(len(types_s.split(',')))))
+    typedargs_s = ''.join([x+" "+y+',' for (x,y) in zip(types_s.split(','),argnames)])[:-1]
+    args_s = ''.join([x+"," for x in argnames])[:-1]
+
+    includes_s = ""
+    for i in args.includefiles.split(','):
+        if "\"" not in i:
+            if i[-2:] != ".h":
+                includes_s = includes_s + "#include <"+i+".h>\n"
+            else:
+                includes_s = includes_s + "#include <"+i+">\n"
+        else:
+            includes_s = includes_s + "#include "+i+"\n"
+
+    # generate tests
+    testruns_s = ''
+    tests = args.arglist.split('),(')
+    print tests
+    for a in tests:
+        t = a.replace('(','').replace(')','')
+        testruns_s += "printf(\""+args.cfunction+" ("+t.replace('"','\\"')+")      %f\\n\", run_test("+args.cfunction+","+t+"));\n"
+
+    print testruns_s
+
+    # Build a C file
+    replace_file(HOST_C_FILE,args.tmpfile,[
+        ("INCLUDES_HERE",includes_s),
+        ("RETTYPE",ret_s),
+        ("RAWTYPES",types_s),
+        ("TYPEDARGS",typedargs_s),
+        ("REALARGS",args_s),
+        ("TESTRUNS",testruns_s)])
+
+    #Build tmpfile
+    gcc_build(args)
+
+
+def asm_snippet(args):
+    print args.asmcode
+    print_unimp("ASM snippets")
+
+
+def getargs():
+    parser = argparse.ArgumentParser(description='Time pretty much anything in C.')
+
+    parser.add_argument('typestring', metavar='typestring', type=str, default=None,
+                        help='Headerfile style typestring')
+
+    parser.add_argument('arglist', metavar='arglist', type=str, default="(0)",
+                        help='List of arguments to try, comma separated tuples. ex: (0,2),(5,3)')
+
+    parser.add_argument('-c', dest='cfunction', action='store',  metavar='C function',
+                        type=str, default=None,
+                        help='A short C snippet. Can use imported functions.')
+
+    parser.add_argument('-a', dest='asmcode', action='store',metavar='ASM snippet',
+                        type=str, default=None,
+                        help='A short x86_64 asm snippet. Can use imported functions.')
+
+    parser.add_argument('-i','--include', dest='includefiles', action='store',metavar='list',
+                        type=str, default="",
+                        help='Comma separated list of #includes required to build')
+
+    parser.add_argument('--tmpfile', dest='tmpfile', action='store',metavar='name',
+                        type=str, default="/tmp/Cats_gen.c",
+                        help='Temporary C file passed into gcc, defaults to in /tmp')
+
+
+    args = parser.parse_args()
+    print args
+    return args
+
+
+args = getargs()
+
+if args.cfunction != None:
+    c_snippet(args)
+elif args.asmcode != None:
+    asm_snippet(args)

timer.c

@@ -0,0 +1,53 @@
+#include <stdio.h>
+#include <stdint.h>
+#include <inttypes.h>
+#include <string.h>
+#include <stdlib.h>
+
+#define PERF_ITRS 2000000
+
+static inline uint64_t rdtscp(){
+  uint64_t v;
+  __asm__ volatile("rdtscp;"
+                   "shl $32,%%rdx;"
+                   "or %%rdx,%%rax;"
+                   : "=a" (v)
+                   :
+                   : "%rcx","%rdx");
+
+  return v;
+}
+
+double run_test(RETTYPE (*function) (RAWTYPES),TYPEDARGS){
+  int ctr = 0;
+  uint8_t real = 0;
+  uint64_t st;
+  uint64_t end;
+  uint64_t offset;
+ runme:
+  st = rdtscp();
+  /* Offset for running the loop and rdtscp */
+  for(ctr=0;ctr<PERF_ITRS;ctr++){
+  }
+  end = rdtscp();
+  offset = end-st;
+
+  st = rdtscp();
+  for(ctr=0;ctr<PERF_ITRS;ctr++){
+    (*function)(REALARGS);
+  }
+  end = rdtscp();
+
+  if(real == 0){ real = 1; goto runme;}
+  /* Run everything for real, previous was just warmup */
+  return (end-st-offset)/(float)PERF_ITRS;
+}
+
+int main(int argc, char* argv[]){
+  printf(    "function ""     cycles\n");
+  printf(    "====================\n");
+
+  TESTRUNS
+
+  return 0;
+}