X-Git-Url: http://git.lttng.org./?a=blobdiff_plain;f=include%2Fust-helper.h;h=e8b393a8114df6333bb6fb4bb6832819cd41c551;hb=7ae0a9445db9a97a43f0054bd38144bfddb00ce0;hp=609a1028642bf946948f9ac10073409c96876d3a;hpb=ddabe860f4cf41a4206a8157d83e6b9354f85cb5;p=lttng-ust.git

diff --git a/include/ust-helper.h b/include/ust-helper.h
index 609a1028..e8b393a8 100644
--- a/include/ust-helper.h
+++ b/include/ust-helper.h
@@ -9,6 +9,8 @@
 
 #include <stdlib.h>
 
+#include <lttng/ust-arch.h>
+
 static inline __attribute__((always_inline))
 void *zmalloc(size_t len)
 {
@@ -37,10 +39,10 @@ void *zmalloc(size_t len)
  * architecture for now by always using the NULL value for the ip
  * context.
  */
-#if defined(__PPC__) && !defined(__PPC64__)
+#if defined(LTTNG_UST_ARCH_PPC) && !defined(LTTNG_UST_ARCH_PPC64)
 #define LTTNG_UST_CALLER_IP()		NULL
-#else /* #if defined(__PPC__) && !defined(__PPC64__) */
+#else
 #define LTTNG_UST_CALLER_IP()		__builtin_return_address(0)
-#endif /* #else #if defined(__PPC__) && !defined(__PPC64__) */
+#endif
 
 #endif /* _LTTNG_UST_HELPER_H */