2 files changed, 21 insertions, 1 deletions

diff --git a/drivers/gpu/nvgpu/include/nvgpu/kref.h b/drivers/gpu/nvgpu/include/nvgpu/kref.h
new file mode 100644
index 00000000..d24db603
--- /dev/null
+++ b/drivers/gpu/nvgpu/include/nvgpu/kref.h
@@ -0,0 +1,20 @@
+/*
+ * Copyright (c) 2017, NVIDIA CORPORATION.  All rights reserved.
+ *
+ * This program is free software; you can redistribute it and/or modify it
+ * under the terms and conditions of the GNU General Public License,
+ * version 2, as published by the Free Software Foundation.
+ *
+ * This program is distributed in the hope it will be useful, but WITHOUT
+ * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
+ * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License for
+ * more details.
+ */
+#ifndef __NVGPU_KREF_H__
+#define __NVGPU_KREF_H__
+
+#ifdef __KERNEL__
+#include <linux/kref.h>
+#endif
+
+#endif /* __NVGPU_KREF_H__ */
diff --git a/drivers/gpu/nvgpu/include/nvgpu/semaphore.h b/drivers/gpu/nvgpu/include/nvgpu/semaphore.h
index 6ab34d67..8e939e72 100644
--- a/drivers/gpu/nvgpu/include/nvgpu/semaphore.h
+++ b/drivers/gpu/nvgpu/include/nvgpu/semaphore.h
@@ -14,7 +14,6 @@
 #ifndef SEMAPHORE_GK20A_H
 #define SEMAPHORE_GK20A_H
 
-#include <linux/kref.h>
 #include <linux/list.h>
 #include <linux/delay.h>
 
@@ -22,6 +21,7 @@
 #include <nvgpu/allocator.h>
 #include <nvgpu/atomic.h>
 #include <nvgpu/bug.h>
+#include <nvgpu/kref.h>
 
 #include "gk20a/gk20a.h"
 #include "gk20a/mm_gk20a.h"