2 files changed, 3 insertions, 1 deletions

diff --git a/drivers/gpu/nvgpu/gk20a/gk20a.c b/drivers/gpu/nvgpu/gk20a/gk20a.c
index f71a51fd..e8db9d2c 100644
--- a/drivers/gpu/nvgpu/gk20a/gk20a.c
+++ b/drivers/gpu/nvgpu/gk20a/gk20a.c
@@ -518,7 +518,8 @@ static void gk20a_free_cb(struct nvgpu_ref *refcount)
         if (g->remove_support)
                 g->remove_support(g);
 
-        kfree(g);
+        if (g->free)
+                g->free(g);
 }
 
 /**
diff --git a/drivers/gpu/nvgpu/gk20a/gk20a.h b/drivers/gpu/nvgpu/gk20a/gk20a.h
index 7e7d9688..5b22d1a4 100644
--- a/drivers/gpu/nvgpu/gk20a/gk20a.h
+++ b/drivers/gpu/nvgpu/gk20a/gk20a.h
@@ -1070,6 +1070,7 @@ struct nvgpu_gpu_params {
 };
 
 struct gk20a {
+        void (*free)(struct gk20a *g);
         struct nvgpu_nvhost_dev *nvhost_dev;
 
         /*