1 files changed, 12 insertions, 4 deletions

diff --git a/drivers/gpu/nvgpu/os/linux/module.c b/drivers/gpu/nvgpu/os/linux/module.c
index 18345ada..5627abfd 100644
--- a/drivers/gpu/nvgpu/os/linux/module.c
+++ b/drivers/gpu/nvgpu/os/linux/module.c
@@ -107,10 +107,7 @@ struct device_node *nvgpu_get_node(struct gk20a *g)
 
 void gk20a_busy_noresume(struct gk20a *g)
 {
-        int ret = pm_runtime_get_if_in_use(dev_from_gk20a(g));
-
-        if (ret <= 0)
-                pm_runtime_get_noresume(dev_from_gk20a(g));
+        pm_runtime_get_noresume(dev_from_gk20a(g));
 }
 
 /*
@@ -262,6 +259,17 @@ int nvgpu_finalize_poweron_linux(struct nvgpu_os_linux *l)
         return 0;
 }
 
+bool gk20a_check_poweron(struct gk20a *g)
+{
+        bool ret;
+
+        nvgpu_mutex_acquire(&g->power_lock);
+        ret = g->power_on;
+        nvgpu_mutex_release(&g->power_lock);
+
+        return ret;
+}
+
 int gk20a_pm_finalize_poweron(struct device *dev)
 {
         struct gk20a *g = get_gk20a(dev);