1 files changed, 1 insertions, 1 deletions

diff --git a/drivers/gpu/nvgpu/gk20a/gk20a.c b/drivers/gpu/nvgpu/gk20a/gk20a.c
index 51b0ba11..8ea2929f 100644
--- a/drivers/gpu/nvgpu/gk20a/gk20a.c
+++ b/drivers/gpu/nvgpu/gk20a/gk20a.c
@@ -379,7 +379,7 @@ void gk20a_driver_start_unload(struct gk20a *g)
         g->driver_is_dying = 1;
         up_write(&g->busy_lock);
 
-        if (gk20a_gpu_is_virtual(g->dev))
+        if (g->is_virtual)
                 return;
 
         gk20a_wait_for_idle(g->dev);