gpu: nvgpu: vgpu: Use vgpu_pm functions in suspend and resume

Use vgpu_pm functions for vgpu instead of gk20a_pm functions
in suspend and resume.

Change-Id: I9d23cd612caa3e6fa9be65d60ccdba3b7f893350
Signed-off-by: Jinyoung Park <jinyoungp@nvidia.com>
Reviewed-on: https://git-master.nvidia.com/r/1544161
Reviewed-by: Thomas Fleury <tfleury@nvidia.com>
Tested-by: Thomas Fleury <tfleury@nvidia.com>
Reviewed-by: svc-mobile-coverity <svc-mobile-coverity@nvidia.com>
Reviewed-by: svccoveritychecker <svccoveritychecker@nvidia.com>
Reviewed-by: Terje Bergstrom <tbergstrom@nvidia.com>
GVS: Gerrit_Virtual_Submit

1 files changed, 8 insertions, 2 deletions

diff --git a/drivers/gpu/nvgpu/common/linux/module.c b/drivers/gpu/nvgpu/common/linux/module.c
index 691fdd69..6a590baa 100644
--- a/drivers/gpu/nvgpu/common/linux/module.c
+++ b/drivers/gpu/nvgpu/common/linux/module.c
@@ -706,7 +706,10 @@ static int gk20a_pm_runtime_resume(struct device *dev)
         if (err)
                 goto fail;
 
-        err = gk20a_pm_finalize_poweron(dev);
+        if (gk20a_gpu_is_virtual(dev))
+                err = vgpu_pm_finalize_poweron(dev);
+        else
+                err = gk20a_pm_finalize_poweron(dev);
         if (err)
                 goto fail_poweron;
 
@@ -722,7 +725,10 @@ static int gk20a_pm_runtime_suspend(struct device *dev)
 {
         int err = 0;
 
-        err = gk20a_pm_prepare_poweroff(dev);
+        if (gk20a_gpu_is_virtual(dev))
+                err = vgpu_pm_prepare_poweroff(dev);
+        else
+                err = gk20a_pm_prepare_poweroff(dev);
         if (err)
                 goto fail;