2 files changed, 3 insertions, 3 deletions

diff --git a/drivers/gpu/nvgpu/gk20a/gk20a_sysfs.c b/drivers/gpu/nvgpu/gk20a/gk20a_sysfs.c
index 8ac983d9..875a99ce 100644
--- a/drivers/gpu/nvgpu/gk20a/gk20a_sysfs.c
+++ b/drivers/gpu/nvgpu/gk20a/gk20a_sysfs.c
@@ -712,7 +712,7 @@ static ssize_t tpc_fs_mask_store(struct device *dev,
 
                 g->ops.gr.set_gpc_tpc_mask(g, 0);
 
-                kfree(g->gr.ctx_vars.local_golden_image);
+                vfree(g->gr.ctx_vars.local_golden_image);
                 g->gr.ctx_vars.local_golden_image = NULL;
                 g->gr.ctx_vars.golden_image_initialized = false;
                 g->gr.ctx_vars.golden_image_size = 0;
diff --git a/drivers/gpu/nvgpu/gk20a/gr_gk20a.c b/drivers/gpu/nvgpu/gk20a/gr_gk20a.c
index eac9456a..883cacdc 100644
--- a/drivers/gpu/nvgpu/gk20a/gr_gk20a.c
+++ b/drivers/gpu/nvgpu/gk20a/gr_gk20a.c
@@ -1680,7 +1680,7 @@ restore_fe_go_idle:
         if (gr->ctx_vars.local_golden_image == NULL) {
 
                 gr->ctx_vars.local_golden_image =
-                        kzalloc(gr->ctx_vars.golden_image_size, GFP_KERNEL);
+                        vzalloc(gr->ctx_vars.golden_image_size);
 
                 if (gr->ctx_vars.local_golden_image == NULL) {
                         err = -ENOMEM;
@@ -3275,7 +3275,7 @@ static void gk20a_remove_gr_support(struct gr_gk20a *gr)
         kfree(gr->ctx_vars.ctxsw_regs.pm_ltc.l);
         kfree(gr->ctx_vars.ctxsw_regs.pm_fbpa.l);
 
-        kfree(gr->ctx_vars.local_golden_image);
+        vfree(gr->ctx_vars.local_golden_image);
         gr->ctx_vars.local_golden_image = NULL;
 
         if (gr->ctx_vars.hwpm_ctxsw_buffer_offset_map)