2 files changed, 4 insertions, 2 deletions

diff --git a/drivers/gpu/nvgpu/gm20b/cde_gm20b.c b/drivers/gpu/nvgpu/gm20b/cde_gm20b.c
index f8267d1d..de7cf872 100644
--- a/drivers/gpu/nvgpu/gm20b/cde_gm20b.c
+++ b/drivers/gpu/nvgpu/gm20b/cde_gm20b.c
@@ -28,15 +28,16 @@ enum programs {
 
 void gm20b_cde_get_program_numbers(struct gk20a *g,
                                           u32 block_height_log2,
+                                          u32 shader_parameter,
                                           int *hprog_out, int *vprog_out)
 {
         int hprog = PROG_HPASS;
         int vprog = (block_height_log2 >= 2) ?
                 PROG_VPASS_LARGE : PROG_VPASS_SMALL;
-        if (g->cde_app.shader_parameter == 1) {
+        if (shader_parameter == 1) {
                 hprog = PROG_PASSTHROUGH;
                 vprog = PROG_PASSTHROUGH;
-        } else if (g->cde_app.shader_parameter == 2) {
+        } else if (shader_parameter == 2) {
                 hprog = PROG_HPASS_DEBUG;
                 vprog = (block_height_log2 >= 2) ?
                         PROG_VPASS_LARGE_DEBUG :
diff --git a/drivers/gpu/nvgpu/gm20b/cde_gm20b.h b/drivers/gpu/nvgpu/gm20b/cde_gm20b.h
index f2ea20a0..0ea423ad 100644
--- a/drivers/gpu/nvgpu/gm20b/cde_gm20b.h
+++ b/drivers/gpu/nvgpu/gm20b/cde_gm20b.h
@@ -20,6 +20,7 @@ struct gk20a;
 
 void gm20b_cde_get_program_numbers(struct gk20a *g,
                                           u32 block_height_log2,
+                                          u32 shader_parameter,
                                           int *hprog_out, int *vprog_out);
 
 #endif