Cleanup: standardise include path

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

diff --git a/tests/benchmark/bench.c b/tests/benchmark/bench.c
index 212baefffefeeb108bfe5a091e9cc9981cdbc708..3fafe5d154207aae70672fbac6251f0d64ca4db2 100644 (file)
--- a/tests/benchmark/bench.c
+++ b/tests/benchmark/bench.c
@@ -20,15 +20,18 @@
  * 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;
 
@@ -44,8 +47,8 @@ void do_stuff(void)
        fclose(file);
        time(NULL);
 
-#ifdef MARKER
-       ust_marker(event, "event %d", v);
+#ifdef TRACING
+       tracepoint(ust_tests_benchmark, tpbench, v);
 #endif
 
 }