Cleanup: standardise include path

[lttng-ust.git] / tests / benchmark / bench.c

diff --git a/tests/benchmark/bench.c b/tests/benchmark/bench.c
index 4e9c355126cb07eab69b62003d64f03afbd29c91..3fafe5d154207aae70672fbac6251f0d64ca4db2 100644 (file)
--- a/tests/benchmark/bench.c
+++ b/tests/benchmark/bench.c
@@ -2,17 +2,36 @@
  * bench.c
  *
  * LTTng Userspace Tracer (UST) - benchmark tool
+ *
+ * Copyright 2010 - Douglas Santos <douglas.santos@polymtl.ca>
+ *
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation; either version 2 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License along
+ * with this program; if not, write to the Free Software Foundation, Inc.,
+ * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  */
 
-#define _GNU_SOURCE
 #include <stdio.h>
 #include <pthread.h>
 #include <stdlib.h>
 #include <unistd.h>
 #include <sched.h>
-#include <ust/marker.h>
 #include <time.h>
 
+#ifdef TRACING
+#define TRACEPOINT_DEFINE
+#include "ust_tests_benchmark.h"
+#endif
+
 static int nr_cpus;
 static unsigned long nr_events;
 
@@ -28,8 +47,8 @@ void do_stuff(void)
        fclose(file);
        time(NULL);
 
-#ifdef MARKER
-       trace_mark(event, "event %d", v);
+#ifdef TRACING
+       tracepoint(ust_tests_benchmark, tpbench, v);
 #endif
 
 }