gpu: nvgpu: gp10b: Fix spill buffer size

Spill buffer size is in chunks of 256B. Multiply the size by
granularity to get the size in bytes.

Bug 1686189

Change-Id: I0462293668322645bd1eab190c12faaeb6c316c1
Signed-off-by: Terje Bergstrom <tbergstrom@nvidia.com>
Reviewed-on: http://git-master/r/801344
(cherry picked from commit 4bf6de7d9c9014a9eaeff56b19437d1841d7cfb0)
Reviewed-on: http://git-master/r/815680

1 files changed, 4 insertions, 2 deletions

diff --git a/drivers/gpu/nvgpu/gp10b/gr_gp10b.c b/drivers/gpu/nvgpu/gp10b/gr_gp10b.c
index c5f45816..61ecddef 100644
--- a/drivers/gpu/nvgpu/gp10b/gr_gp10b.c
+++ b/drivers/gpu/nvgpu/gp10b/gr_gp10b.c
@@ -510,7 +510,8 @@ static int gr_gp10b_alloc_gr_ctx(struct gk20a *g,
 
         if (flags & NVGPU_ALLOC_OBJ_FLAGS_GFXP) {
                 u32 spill_size =
-                        gr_gpc0_swdx_rm_spill_buffer_size_256b_default_v();
+                        gr_gpc0_swdx_rm_spill_buffer_size_256b_default_v() *
+                        gr_gpc0_swdx_rm_spill_buffer_size_256b_byte_granularity_v();
                 u32 pagepool_size = g->ops.gr.pagepool_default_size(g) *
                         gr_scc_pagepool_total_pages_byte_granularity_v();
                 u32 betacb_size = g->gr.attrib_cb_default_size +
@@ -694,7 +695,8 @@ static void gr_gp10b_update_ctxsw_preemption_mode(struct gk20a *g,
                         gr_gpc0_swdx_rm_spill_buffer_addr_39_8_align_bits_v()) |
                         (u64_hi32(gr_ctx->t18x.spill_ctxsw_buffer.gpu_va) <<
                          (32 - gr_gpc0_swdx_rm_spill_buffer_addr_39_8_align_bits_v()));
-                size = gr_ctx->t18x.spill_ctxsw_buffer.size;
+                size = gr_ctx->t18x.spill_ctxsw_buffer.size /
+                        gr_gpc0_swdx_rm_spill_buffer_size_256b_byte_granularity_v();
 
                 gr_gk20a_ctx_patch_write(g, ch_ctx,
                                 gr_gpc0_swdx_rm_spill_buffer_addr_r(),