2 files changed, 4 insertions, 0 deletions

diff --git a/drivers/gpu/nvgpu/gv11b/gr_gv11b.c b/drivers/gpu/nvgpu/gv11b/gr_gv11b.c
index df323da1..84699db7 100644
--- a/drivers/gpu/nvgpu/gv11b/gr_gv11b.c
+++ b/drivers/gpu/nvgpu/gv11b/gr_gv11b.c
@@ -1235,6 +1235,9 @@ int gr_gv11b_handle_sw_method(struct gk20a *g, u32 addr,
                 case NVC3C0_SET_SKEDCHECK:
                         gr_gv11b_set_skedcheck(g, data);
                         break;
+                case NVC3C0_SET_SHADER_CUT_COLLECTOR:
+                        gr_gv11b_set_shader_cut_collector(g, data);
+                        break;
                 default:
                         goto fail;
                 }
diff --git a/drivers/gpu/nvgpu/gv11b/gr_gv11b.h b/drivers/gpu/nvgpu/gv11b/gr_gv11b.h
index dac82677..b4a7e411 100644
--- a/drivers/gpu/nvgpu/gv11b/gr_gv11b.h
+++ b/drivers/gpu/nvgpu/gv11b/gr_gv11b.h
@@ -78,6 +78,7 @@ enum {
 #define NVC397_SET_SHADER_CUT_COLLECTOR_STATE_ENABLE            0x1
 
 #define NVC3C0_SET_SKEDCHECK                    0x23c
+#define NVC3C0_SET_SHADER_CUT_COLLECTOR         0x250
 
 #define NVA297_SET_SHADER_EXCEPTIONS_ENABLE_FALSE 0