1 files changed, 20 insertions, 1 deletions

diff --git a/drivers/gpu/nvgpu/gp106/fb_gp106.c b/drivers/gpu/nvgpu/gp106/fb_gp106.c
index 1c5b3e0a..ef9f1094 100644
--- a/drivers/gpu/nvgpu/gp106/fb_gp106.c
+++ b/drivers/gpu/nvgpu/gp106/fb_gp106.c
@@ -12,14 +12,33 @@
  */
 
 #include <linux/types.h>
+#include <linux/delay.h>
 
 #include "gk20a/gk20a.h"
 #include "gp10b/fb_gp10b.h"
+#include "hw_fb_gp106.h"
+
+#define HW_SCRUB_TIMEOUT_DEFAULT        100 /* usec */
+#define HW_SCRUB_TIMEOUT_MAX            2000000 /* usec */
+
+static void gp106_fb_reset(struct gk20a *g)
+{
+        int retries = HW_SCRUB_TIMEOUT_MAX / HW_SCRUB_TIMEOUT_DEFAULT;
+        /* wait for memory to be accessible */
+        do {
+                u32 w = gk20a_readl(g, fb_niso_scrub_status_r());
+                if (fb_niso_scrub_status_flag_v(w)) {
+                        gk20a_dbg_fn("done");
+                        break;
+                }
+                udelay(HW_SCRUB_TIMEOUT_DEFAULT);
+        } while (--retries);
+}
 
 void gp106_init_fb(struct gpu_ops *gops)
 {
         gp10b_init_fb(gops);
 
         gops->fb.init_fs_state = NULL;
-        gops->fb.reset = NULL;
+        gops->fb.reset = gp106_fb_reset;
 }