1 files changed, 2 insertions, 1 deletions

diff --git a/drivers/gpu/nvgpu/pci.c b/drivers/gpu/nvgpu/pci.c
index 7ef626c2..0ed621ce 100644
--- a/drivers/gpu/nvgpu/pci.c
+++ b/drivers/gpu/nvgpu/pci.c
@@ -481,7 +481,8 @@ static void nvgpu_pci_remove(struct pci_dev *pdev)
 
         enable_irq(g->irq_stall);
 
-        kfree(g);
+        gk20a_get_platform(&pdev->dev)->g = NULL;
+        gk20a_put(g);
 }
 
 static struct pci_driver nvgpu_pci_driver = {