1 files changed, 8 insertions, 0 deletions

diff --git a/drivers/gpu/nvgpu/gp10b/mm_gp10b.h b/drivers/gpu/nvgpu/gp10b/mm_gp10b.h
index 034944e0..f34200a0 100644
--- a/drivers/gpu/nvgpu/gp10b/mm_gp10b.h
+++ b/drivers/gpu/nvgpu/gp10b/mm_gp10b.h
@@ -14,8 +14,16 @@
 #ifndef MM_GP10B_H
 #define MM_GP10B_H
 
+#include <linux/version.h>
+
 #define NVGPU_MM_GET_IO_COHERENCE_BIT   35
 
+#if LINUX_VERSION_CODE < KERNEL_VERSION(3, 18, 0)
+#define NVGPU_USE_NEW_ALLOCATOR 1
+#else
+#define NVGPU_USE_NEW_ALLOCATOR 0
+#endif
+
 struct gpu_ops;
 
 void gp10b_init_mm(struct gpu_ops *gops);