1 files changed, 5 insertions, 0 deletions

diff --git a/drivers/gpu/nvgpu/os/linux/vgpu/gv11b/platform_gv11b_vgpu_tegra.c b/drivers/gpu/nvgpu/os/linux/vgpu/gv11b/platform_gv11b_vgpu_tegra.c
index 66911626..8eada9df 100644
--- a/drivers/gpu/nvgpu/os/linux/vgpu/gv11b/platform_gv11b_vgpu_tegra.c
+++ b/drivers/gpu/nvgpu/os/linux/vgpu/gv11b/platform_gv11b_vgpu_tegra.c
@@ -22,6 +22,7 @@
 #include "os/linux/vgpu/clk_vgpu.h"
 #include "os/linux/platform_gk20a.h"
 #include "os/linux/os_linux.h"
+#include "os/linux/vgpu/vgpu_linux.h"
 
 static int gv11b_vgpu_probe(struct device *dev)
 {
@@ -94,4 +95,8 @@ struct gk20a_platform gv11b_vgpu_tegra_platform = {
         .devfreq_governor = "userspace",
 
         .virtual_dev = true,
+
+        /* power management callbacks */
+        .suspend = vgpu_tegra_suspend,
+        .resume = vgpu_tegra_resume,
 };