Drop 'linux/atomic.h' wrapper

[lttng-modules.git] / include / wrapper / barrier.h

diff --git a/include/wrapper/barrier.h b/include/wrapper/barrier.h
index c2f0545b4bc93ded40047c770ca99543cd8d2bc6..9d47ad7e5fa1939be5272a9c003b02536723cc00 100644 (file)
--- a/include/wrapper/barrier.h
+++ b/include/wrapper/barrier.h
@@ -10,10 +10,14 @@
 #ifndef _LTTNG_WRAPPER_BARRIER_H
 #define _LTTNG_WRAPPER_BARRIER_H
 
-#include <linux/version.h>
+#include <lttng/kernel-version.h>
+#if LTTNG_LINUX_VERSION_CODE >= LTTNG_KERNEL_VERSION(3,4,0)
 #include <asm/barrier.h>
+#else
+#include <asm/system.h>
+#endif
 
-#if LINUX_VERSION_CODE >= KERNEL_VERSION(3,14,0)
+#if LTTNG_LINUX_VERSION_CODE >= LTTNG_KERNEL_VERSION(3,14,0)
 
 #define lttng_smp_store_release(x, v) smp_store_release(x, v)
 #define lttng_smp_load_acquire(x) smp_load_acquire(x)