2 files changed, 3 insertions, 0 deletions

diff --git a/drivers/gpu/nvgpu/gk20a/gk20a.h b/drivers/gpu/nvgpu/gk20a/gk20a.h
index 441ea7cc..4fac3cc5 100644
--- a/drivers/gpu/nvgpu/gk20a/gk20a.h
+++ b/drivers/gpu/nvgpu/gk20a/gk20a.h
@@ -176,6 +176,7 @@ struct gpu_ops {
                 void (*init_kind_attr)(struct gk20a *g);
                 void (*set_mmu_page_size)(struct gk20a *g);
                 int (*compression_page_size)(struct gk20a *g);
+                void (*dump_vpr_wpr_info)(struct gk20a *g);
         } fb;
         struct {
                 void (*slcg_bus_load_gating_prod)(struct gk20a *g, bool prod);
diff --git a/drivers/gpu/nvgpu/gk20a/mm_gk20a.c b/drivers/gpu/nvgpu/gk20a/mm_gk20a.c
index 67680c4c..b53d53a3 100644
--- a/drivers/gpu/nvgpu/gk20a/mm_gk20a.c
+++ b/drivers/gpu/nvgpu/gk20a/mm_gk20a.c
@@ -2805,6 +2805,8 @@ int gk20a_mm_fb_flush(struct gk20a *g)
         if (tegra_platform_is_silicon() && retry < 0) {
                 gk20a_warn(dev_from_gk20a(g),
                         "fb_flush too many retries");
+                if (g->ops.fb.dump_vpr_wpr_info)
+                        g->ops.fb.dump_vpr_wpr_info(g);
                 ret = -EBUSY;
         }