1 files changed, 2 insertions, 0 deletions

diff --git a/drivers/gpu/nvgpu/include/nvgpu/linux/atomic.h b/drivers/gpu/nvgpu/include/nvgpu/linux/atomic.h
index c6dd4650..1fdb2674 100644
--- a/drivers/gpu/nvgpu/include/nvgpu/linux/atomic.h
+++ b/drivers/gpu/nvgpu/include/nvgpu/linux/atomic.h
@@ -16,7 +16,9 @@
 #ifndef __NVGPU_ATOMIC_LINUX_H__
 #define __NVGPU_ATOMIC_LINUX_H__
 
+#ifdef __KERNEL__
 #include <linux/atomic.h>
+#endif
 
 typedef struct nvgpu_atomic {
   atomic_t atomic_var;