diff --git a/drivers/gpu/nvgpu/gm20b/gr_gm20b.c b/drivers/gpu/nvgpu/gm20b/gr_gm20b.c
index 20eaae5fc..e1204dad9 100644
--- a/drivers/gpu/nvgpu/gm20b/gr_gm20b.c
+++ b/drivers/gpu/nvgpu/gm20b/gr_gm20b.c
@@ -528,7 +528,7 @@ static void gr_gm20b_set_gpc_tpc_mask(struct gk20a *g, u32 gpc_index)
 	}
 }
 
-static int gr_gm20b_ctx_state_floorsweep(struct gk20a *g)
+int gr_gm20b_ctx_state_floorsweep(struct gk20a *g)
 {
 	struct gr_gk20a *gr = &g->gr;
 	u32 tpc_index, gpc_index;
diff --git a/drivers/gpu/nvgpu/gm20b/gr_gm20b.h b/drivers/gpu/nvgpu/gm20b/gr_gm20b.h
index 646c16d94..90f933bd3 100644
--- a/drivers/gpu/nvgpu/gm20b/gr_gm20b.h
+++ b/drivers/gpu/nvgpu/gm20b/gr_gm20b.h
@@ -46,4 +46,5 @@ void gm20b_init_gr(struct gpu_ops *gops);
 void gr_gm20b_commit_global_attrib_cb(struct gk20a *g,
 				      struct channel_ctx_gk20a *ch_ctx,
 				      u64 addr, bool patch);
+int gr_gm20b_ctx_state_floorsweep(struct gk20a *g);
 #endif