1 files changed, 5 insertions, 2 deletions

diff --git a/drivers/gpu/nvgpu/gk20a/mm_gk20a.c b/drivers/gpu/nvgpu/gk20a/mm_gk20a.c
index b642075e..644531f1 100644
--- a/drivers/gpu/nvgpu/gk20a/mm_gk20a.c
+++ b/drivers/gpu/nvgpu/gk20a/mm_gk20a.c
@@ -468,8 +468,11 @@ int gk20a_mm_fb_flush(struct gk20a *g)
         } while (!nvgpu_timeout_expired(&timeout));
 
         if (nvgpu_timeout_peek_expired(&timeout)) {
-                if (g->ops.fb.dump_vpr_wpr_info) {
-                        g->ops.fb.dump_vpr_wpr_info(g);
+                if (g->ops.fb.dump_vpr_info) {
+                        g->ops.fb.dump_vpr_info(g);
+                }
+                if (g->ops.fb.dump_wpr_info) {
+                        g->ops.fb.dump_wpr_info(g);
                 }
                 ret = -EBUSY;
         }