gpu: nvgpu: Reset FB also in gm20b

FB reset was added for gk20a. It should be invoked also on gm20b.

Change-Id: I0b074bc50a889108edae93d62b3194e54bfda881
Signed-off-by: Terje Bergstrom <tbergstrom@nvidia.com>
Reviewed-on: http://git-master/r/765366

3 files changed, 4 insertions, 1 deletions

diff --git a/drivers/gpu/nvgpu/gk20a/fb_gk20a.c b/drivers/gpu/nvgpu/gk20a/fb_gk20a.c
index 35551939..2966e654 100644
--- a/drivers/gpu/nvgpu/gk20a/fb_gk20a.c
+++ b/drivers/gpu/nvgpu/gk20a/fb_gk20a.c
@@ -20,7 +20,7 @@
 #include "hw_mc_gk20a.h"
 #include "hw_fb_gk20a.h"
 
-static void fb_gk20a_reset(struct gk20a *g)
+void fb_gk20a_reset(struct gk20a *g)
 {
         u32 val;
 
diff --git a/drivers/gpu/nvgpu/gk20a/fb_gk20a.h b/drivers/gpu/nvgpu/gk20a/fb_gk20a.h
index 49dd5fd7..e5ea9e42 100644
--- a/drivers/gpu/nvgpu/gk20a/fb_gk20a.h
+++ b/drivers/gpu/nvgpu/gk20a/fb_gk20a.h
@@ -16,4 +16,5 @@
 struct gk20a;
 
 void gk20a_init_fb(struct gpu_ops *gops);
+void fb_gk20a_reset(struct gk20a *g);
 #endif
diff --git a/drivers/gpu/nvgpu/gm20b/fb_gm20b.c b/drivers/gpu/nvgpu/gm20b/fb_gm20b.c
index 746ed42f..3d4604c5 100644
--- a/drivers/gpu/nvgpu/gm20b/fb_gm20b.c
+++ b/drivers/gpu/nvgpu/gm20b/fb_gm20b.c
@@ -17,6 +17,7 @@
 
 #include "gk20a/gk20a.h"
 #include "gk20a/kind_gk20a.h"
+#include "gk20a/fb_gk20a.h"
 
 #include "hw_fb_gm20b.h"
 #include "hw_top_gm20b.h"
@@ -131,6 +132,7 @@ static void gm20b_fb_dump_vpr_wpr_info(struct gk20a *g)
 
 void gm20b_init_fb(struct gpu_ops *gops)
 {
+        gops->fb.reset = fb_gk20a_reset;
         gops->fb.init_fs_state = fb_gm20b_init_fs_state;
         gops->fb.set_mmu_page_size = gm20b_fb_set_mmu_page_size;
         gops->fb.compression_page_size = gm20b_fb_compression_page_size;