1 files changed, 1 insertions, 1 deletions

diff --git a/drivers/gpu/nvgpu/gk20a/ce2_gk20a.c b/drivers/gpu/nvgpu/gk20a/ce2_gk20a.c
index cce7a769..7aae2584 100644
--- a/drivers/gpu/nvgpu/gk20a/ce2_gk20a.c
+++ b/drivers/gpu/nvgpu/gk20a/ce2_gk20a.c
@@ -436,7 +436,7 @@ u32 gk20a_ce_create_context_with_cb(struct device *dev,
 
         err = nvgpu_mutex_init(&ce_ctx->gpu_ctx_mutex);
         if (err) {
-                kfree(ce_ctx);
+                nvgpu_kfree(g, ce_ctx);
                 return ctx_id;
         }