1 files changed, 2 insertions, 0 deletions

diff --git a/drivers/gpu/nvgpu/gk20a/gk20a.c b/drivers/gpu/nvgpu/gk20a/gk20a.c
index f2094bfa..5f365d4b 100644
--- a/drivers/gpu/nvgpu/gk20a/gk20a.c
+++ b/drivers/gpu/nvgpu/gk20a/gk20a.c
@@ -1390,6 +1390,8 @@ static void gk20a_pm_shutdown(struct platform_device *pdev)
         /* Prevent more requests by disabling Runtime PM */
         __pm_runtime_disable(&pdev->dev, false);
 
+        gk20a_wait_for_idle(&pdev->dev);
+
         /* Be ready for rail-gate after this point */
         if (gk20a_gpu_is_virtual(&pdev->dev))
                 vgpu_pm_prepare_poweroff(&pdev->dev);