1 files changed, 1 insertions, 3 deletions

diff --git a/drivers/gpu/nvgpu/gk20a/ce2_gk20a.c b/drivers/gpu/nvgpu/gk20a/ce2_gk20a.c
index af49c864..7afed41f 100644
--- a/drivers/gpu/nvgpu/gk20a/ce2_gk20a.c
+++ b/drivers/gpu/nvgpu/gk20a/ce2_gk20a.c
@@ -453,9 +453,7 @@ u32 gk20a_ce_create_context_with_cb(struct device *dev,
          }
 
         /* bind the channel to the vm */
-        gk20a_vm_get(&g->mm.ce.vm);
-        ce_ctx->vm = ce_ctx->ch->vm = &g->mm.ce.vm;
-        err = channel_gk20a_commit_va(ce_ctx->ch);
+        err = __gk20a_vm_bind_channel(&g->mm.ce.vm, ce_ctx->ch);
         if (err) {
                 gk20a_err(ce_ctx->dev, "ce: could not bind vm");
                 goto end;