1 files changed, 1 insertions, 1 deletions

diff --git a/drivers/gpu/nvgpu/gp10b/gr_gp10b.c b/drivers/gpu/nvgpu/gp10b/gr_gp10b.c
index 33a52db5..240bab81 100644
--- a/drivers/gpu/nvgpu/gp10b/gr_gp10b.c
+++ b/drivers/gpu/nvgpu/gp10b/gr_gp10b.c
@@ -508,7 +508,7 @@ static int gr_gp10b_alloc_gr_ctx(struct gk20a *g,
                         goto fail_free_preempt;
                 }
 
-                err = gk20a_gmmu_alloc_map(vm, betacb_size,
+                err = gk20a_gmmu_alloc_map(vm, attrib_cb_size,
                                            &(*gr_ctx)->t18x.betacb_ctxsw_buffer);
                 if (err) {
                         gk20a_err(dev_from_gk20a(vm->mm->g),