ppc fix

[lttv.git] / ltt-usertrace / ltt / ltt-usertrace.h

diff --git a/ltt-usertrace/ltt/ltt-usertrace.h b/ltt-usertrace/ltt/ltt-usertrace.h
index 7770f9580cd93a1a61c382b5b86074a131edcc32..90729c679f6e964fc2f1208d18e9046b14b0bfc4 100644 (file)
--- a/ltt-usertrace/ltt/ltt-usertrace.h
+++ b/ltt-usertrace/ltt/ltt-usertrace.h
@@ -11,17 +11,14 @@
 
 #include <errno.h>
 #include <syscall.h>
-#include <linux/unistd.h>
 #include <string.h>
-#include <sys/types.h>
 #include <stdint.h>
-#ifdef powerpc
-#define __KERNEL__     /* Ugly hack : atomic.h is broken */
-#endif
-#include <asm/atomic.h>
+#include <sys/types.h>
+#include <linux/unistd.h>
+#if defined(__powerpc__) || defined(__powerpc64__)
 #include "ltt/ltt-usertrace-ppc.h"
-#ifdef powerpc
-#undef __KERNEL__      /* Ugly hack : atomic.h is broken */
+#else
+#include <asm/atomic.h>
 #endif
 
 #ifndef min