1 files changed, 3 insertions, 0 deletions

diff --git a/drivers/gpu/nvgpu/gk20a/gk20a.c b/drivers/gpu/nvgpu/gk20a/gk20a.c
index badb7717..3dfc50e8 100644
--- a/drivers/gpu/nvgpu/gk20a/gk20a.c
+++ b/drivers/gpu/nvgpu/gk20a/gk20a.c
@@ -1148,6 +1148,9 @@ static void gk20a_pm_shutdown(struct platform_device *pdev)
                         break;
         }
 #endif
+
+        /* Be ready for rail-gate after this point */
+        gk20a_pm_prepare_poweroff(&pdev->dev);
 }
 
 #ifdef CONFIG_PM