diff --git a/drivers/gpu/nvgpu/common/linux/driver_common.c b/drivers/gpu/nvgpu/common/linux/driver_common.c
index b3333e371..9029366ff 100644
--- a/drivers/gpu/nvgpu/common/linux/driver_common.c
+++ b/drivers/gpu/nvgpu/common/linux/driver_common.c
@@ -14,6 +14,7 @@
  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
  */
 
+#include <linux/reboot.h>
 #include <linux/dma-mapping.h>
 #include <linux/mm.h>
 #include <uapi/linux/nvgpu.h>
@@ -37,6 +38,11 @@
 
 #define EMC3D_DEFAULT_RATIO 750
 
+void nvgpu_kernel_restart(void *cmd)
+{
+	kernel_restart(cmd);
+}
+
 static void nvgpu_init_vars(struct gk20a *g)
 {
 	struct nvgpu_os_linux *l = nvgpu_os_linux_from_gk20a(g);
diff --git a/drivers/gpu/nvgpu/gk20a/gk20a.c b/drivers/gpu/nvgpu/gk20a/gk20a.c
index 868792c0b..6aae851e0 100644
--- a/drivers/gpu/nvgpu/gk20a/gk20a.c
+++ b/drivers/gpu/nvgpu/gk20a/gk20a.c
@@ -22,8 +22,6 @@
  * DEALINGS IN THE SOFTWARE.
  */
 
-#include <linux/reboot.h>
-
 #include <nvgpu/nvgpu_common.h>
 #include <nvgpu/kmem.h>
 #include <nvgpu/allocator.h>
@@ -61,7 +59,7 @@ void __nvgpu_check_gpu_state(struct gk20a *g)
 	if (boot_0 == 0xffffffff) {
 		nvgpu_err(g, "GPU has disappeared from bus!!");
 		nvgpu_err(g, "Rebooting system!!");
-		kernel_restart(NULL);
+		nvgpu_kernel_restart(NULL);
 	}
 }
 
diff --git a/drivers/gpu/nvgpu/include/nvgpu/nvgpu_common.h b/drivers/gpu/nvgpu/include/nvgpu/nvgpu_common.h
index 50e8c8f15..34660516b 100644
--- a/drivers/gpu/nvgpu/include/nvgpu/nvgpu_common.h
+++ b/drivers/gpu/nvgpu/include/nvgpu/nvgpu_common.h
@@ -31,4 +31,6 @@ int nvgpu_probe(struct gk20a *g,
 		const char *interface_name,
 		struct class *class);
 
+void nvgpu_kernel_restart(void *cmd);
+
 #endif