X-Git-Url: http://git.liburcu.org/?a=blobdiff_plain;f=wrapper%2Fatomic.h;h=e0e899bdf075a313903eada317d084b52fa0b019;hb=78f56ae3a0c007bf1be7e01731d372f79fe6781e;hp=51920c4a03fe673a36edb4c3b70d71e18614dcd8;hpb=9f36eaed6f91d5897924b551b44d1edd8cee00e2;p=lttng-modules.git

diff --git a/wrapper/atomic.h b/wrapper/atomic.h
index 51920c4a..e0e899bd 100644
--- a/wrapper/atomic.h
+++ b/wrapper/atomic.h
@@ -10,14 +10,10 @@
 #ifndef _LTTNG_WRAPPER_ATOMIC_H
 #define _LTTNG_WRAPPER_ATOMIC_H
 
-#include <linux/version.h>
-#if (LINUX_VERSION_CODE >= KERNEL_VERSION(2,6,37))
+#include <lttng-kernel-version.h>
 #include <linux/atomic.h>
-#else
-#include <asm/atomic.h>
-#endif
 
-#if (LINUX_VERSION_CODE >= KERNEL_VERSION(3,16,0))
+#if (LTTNG_LINUX_VERSION_CODE >= LTTNG_KERNEL_VERSION(3,16,0))
 static inline void lttng_smp_mb__before_atomic(void)
 {
 	smp_mb__before_atomic();
@@ -27,7 +23,7 @@ static inline void lttng_smp_mb__after_atomic(void)
 {
 	smp_mb__after_atomic();
 }
-#else /* #if (LINUX_VERSION_CODE >= KERNEL_VERSION(3,16,0)) */
+#else /* #if (LTTNG_LINUX_VERSION_CODE >= LTTNG_KERNEL_VERSION(3,16,0)) */
 static inline void lttng_smp_mb__before_atomic(void)
 {
 	smp_mb__before_atomic_inc();
@@ -37,6 +33,6 @@ static inline void lttng_smp_mb__after_atomic(void)
 {
 	smp_mb__after_atomic_inc();
 }
-#endif /* #else #if (LINUX_VERSION_CODE >= KERNEL_VERSION(3,16,0)) */
+#endif /* #else #if (LTTNG_LINUX_VERSION_CODE >= LTTNG_KERNEL_VERSION(3,16,0)) */
 
 #endif /* _LTTNG_WRAPPER_ATOMIC_H */