1 files changed, 2 insertions, 1 deletions

diff --git a/drivers/gpu/nvgpu/gk20a/gr_gk20a.c b/drivers/gpu/nvgpu/gk20a/gr_gk20a.c
index d568b1bc..524547e7 100644
--- a/drivers/gpu/nvgpu/gk20a/gr_gk20a.c
+++ b/drivers/gpu/nvgpu/gk20a/gr_gk20a.c
@@ -337,7 +337,8 @@ static int gr_gk20a_wait_fe_idle(struct gk20a *g, unsigned long end_jiffies,
                 val = gk20a_readl(g, gr_status_r());
 
                 if (!gr_status_fe_method_upper_v(val) &&
-                        !gr_status_fe_method_lower_v(val)) {
+                        !gr_status_fe_method_lower_v(val) &&
+                        !gr_status_fe_gi_v(val)) {
                         gk20a_dbg_fn("done");
                         return 0;
                 }