X-Git-Url: http://git.liburcu.org/?a=blobdiff_plain;f=wrapper%2Fatomic.h;h=224814fbc59e60d57d6ef98f73f582b866607df4;hb=360d3efee55154a55c18be32175bee608dcb21df;hp=6a3cd076b28bb05e7695bb9aef1c205c387cdda3;hpb=505fb41059f97c7c4ee456bafabcb6528df8ebe2;p=lttng-modules.git

diff --git a/wrapper/atomic.h b/wrapper/atomic.h
index 6a3cd076..224814fb 100644
--- a/wrapper/atomic.h
+++ b/wrapper/atomic.h
@@ -23,10 +23,14 @@
  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  */
 
-#include <linux/version.h>
+#include <lttng-kernel-version.h>
+#if (LTTNG_LINUX_VERSION_CODE >= LTTNG_KERNEL_VERSION(2,6,37))
 #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();
@@ -36,7 +40,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();
@@ -46,6 +50,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 */