1 files changed, 6 insertions, 1 deletions

diff --git a/drivers/gpu/nvgpu/os/linux/vgpu/platform_vgpu_tegra.c b/drivers/gpu/nvgpu/os/linux/vgpu/platform_vgpu_tegra.c
index e4819e7d..44879a45 100644
--- a/drivers/gpu/nvgpu/os/linux/vgpu/platform_vgpu_tegra.c
+++ b/drivers/gpu/nvgpu/os/linux/vgpu/platform_vgpu_tegra.c
@@ -1,7 +1,7 @@
 /*
  * Tegra Virtualized GPU Platform Interface
  *
- * Copyright (c) 2014-2017, NVIDIA CORPORATION.  All rights reserved.
+ * Copyright (c) 2014-2018, 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,
@@ -21,6 +21,7 @@
 #include "gk20a/gk20a.h"
 #include "os/linux/platform_gk20a.h"
 #include "clk_vgpu.h"
+#include "vgpu_linux.h"
 
 static int gk20a_tegra_probe(struct device *dev)
 {
@@ -66,4 +67,8 @@ struct gk20a_platform vgpu_tegra_platform = {
         .devfreq_governor = "userspace",
 
         .virtual_dev = true,
+
+        /* power management callbacks */
+        .suspend = vgpu_tegra_suspend,
+        .resume = vgpu_tegra_resume,
 };