3 files changed, 9 insertions, 3 deletions

diff --git a/drivers/gpu/nvgpu/common/linux/driver_common.c b/drivers/gpu/nvgpu/common/linux/driver_common.c
index b3333e37..9029366f 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 868792c0..6aae851e 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 50e8c8f1..34660516 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