gpu: nvgpu: gp10b: Fix CB size for GfxP

Program correct CB size for GfxP channels. We were accidentally
using the context image size.

Change-Id: I273215256e41e89b7d76f3294a73641804beeb79
Signed-off-by: Terje Bergstrom <tbergstrom@nvidia.com>
Reviewed-on: http://git-master/r/792713
Reviewed-on: http://git-master/r/806188

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 e727ee99..b614da64 100644
--- a/drivers/gpu/nvgpu/gp10b/gr_gp10b.c
+++ b/drivers/gpu/nvgpu/gp10b/gr_gp10b.c
@@ -902,7 +902,7 @@ static void gr_gp10b_commit_global_attrib_cb(struct gk20a *g,
         int attrBufferSize;
 
         if (gr_ctx->t18x.preempt_ctxsw_buffer.gpu_va)
-                attrBufferSize = gr_ctx->t18x.preempt_ctxsw_buffer.size;
+                attrBufferSize = gr_ctx->t18x.betacb_ctxsw_buffer.size;
         else
                 attrBufferSize = g->ops.gr.calc_global_ctx_buffer_size(g);