gpu: nvgpu: gp10b: Calc global context buffer size

In gp10b we need to limit global context buffer size, and it needs
to be 128b aligned.

Change-Id: I51570e2457a374c09be4d611e683ae30917f9fc0
Signed-off-by: Terje Bergstrom <tbergstrom@nvidia.com>
Reviewed-on: http://git-master/r/657911
GVS: Gerrit_Virtual_Submit

1 files changed, 30 insertions, 0 deletions

diff --git a/drivers/gpu/nvgpu/gp10b/gr_gp10b.c b/drivers/gpu/nvgpu/gp10b/gr_gp10b.c
index 7a316106..b8fbc28e 100644
--- a/drivers/gpu/nvgpu/gp10b/gr_gp10b.c
+++ b/drivers/gpu/nvgpu/gp10b/gr_gp10b.c
@@ -305,6 +305,34 @@ static void gr_gp10b_buffer_size_defaults(struct gk20a *g)
                 gr_scc_pagepool_total_pages_hwmax_value_v();
 }
 
+static int gr_gp10b_calc_global_ctx_buffer_size(struct gk20a *g)
+{
+        struct gr_gk20a *gr = &g->gr;
+        int size;
+
+        gr->attrib_cb_size = gr->attrib_cb_default_size
+                + (gr->attrib_cb_default_size >> 1);
+        gr->alpha_cb_size = gr->alpha_cb_default_size
+                + (gr->alpha_cb_default_size >> 1);
+
+        gr->attrib_cb_size = min(gr->attrib_cb_size,
+                                 gr_gpc0_ppc0_cbm_beta_cb_size_v_f(0xffffffff));
+        gr->alpha_cb_size = min(gr->attrib_cb_size,
+                                 gr_gpc0_ppc0_cbm_alpha_cb_size_v_f(0xffffffff));
+
+        size = gr->attrib_cb_size *
+                gr_gpc0_ppc0_cbm_beta_cb_size_v_granularity_v() *
+                gr->max_tpc_count;
+
+        size += gr->alpha_cb_size *
+                gr_gpc0_ppc0_cbm_alpha_cb_size_v_granularity_v() *
+                gr->max_tpc_count;
+
+        size = ALIGN(size, 128);
+
+        return size;
+}
+
 void gp10b_init_gr(struct gpu_ops *gops)
 {
         gm20b_init_gr(gops);
@@ -314,4 +342,6 @@ void gp10b_init_gr(struct gpu_ops *gops)
         gops->gr.add_zbc_color = gr_gp10b_add_zbc_color;
         gops->gr.add_zbc_depth = gr_gp10b_add_zbc_depth;
         gops->gr.buffer_size_defaults = gr_gp10b_buffer_size_defaults;
+        gops->gr.calc_global_ctx_buffer_size =
+                gr_gp10b_calc_global_ctx_buffer_size;
 }