2 files changed, 4 insertions, 2 deletions

diff --git a/drivers/gpu/nvgpu/gp10b/cde_gp10b.c b/drivers/gpu/nvgpu/gp10b/cde_gp10b.c
index 685ddbc4..1ddbcba6 100644
--- a/drivers/gpu/nvgpu/gp10b/cde_gp10b.c
+++ b/drivers/gpu/nvgpu/gp10b/cde_gp10b.c
@@ -32,17 +32,18 @@ enum gp10b_programs {
 
 void gp10b_cde_get_program_numbers(struct gk20a *g,
                                           u32 block_height_log2,
+                                          u32 shader_parameter,
                                           int *hprog_out, int *vprog_out)
 {
         int hprog, vprog;
 
-        if (g->cde_app.shader_parameter == 1) {
+        if (shader_parameter == 1) {
                 hprog = GP10B_PROG_PASSTHROUGH;
                 vprog = GP10B_PROG_PASSTHROUGH;
         } else {
                 hprog = GP10B_PROG_HPASS;
                 vprog = GP10B_PROG_VPASS;
-                if (g->cde_app.shader_parameter == 2) {
+                if (shader_parameter == 2) {
                         hprog = GP10B_PROG_HPASS_DEBUG;
                         vprog = GP10B_PROG_VPASS_DEBUG;
                 }
diff --git a/drivers/gpu/nvgpu/gp10b/cde_gp10b.h b/drivers/gpu/nvgpu/gp10b/cde_gp10b.h
index 3ee6027c..7ccfe560 100644
--- a/drivers/gpu/nvgpu/gp10b/cde_gp10b.h
+++ b/drivers/gpu/nvgpu/gp10b/cde_gp10b.h
@@ -21,6 +21,7 @@ struct sg_table;
 
 void gp10b_cde_get_program_numbers(struct gk20a *g,
                                           u32 block_height_log2,
+                                          u32 shader_parameter,
                                           int *hprog_out, int *vprog_out);
 bool gp10b_need_scatter_buffer(struct gk20a *g);
 int gp10b_populate_scatter_buffer(struct gk20a *g,