1 files changed, 6 insertions, 7 deletions

diff --git a/drivers/gpu/nvgpu/gv11b/fifo_gv11b.c b/drivers/gpu/nvgpu/gv11b/fifo_gv11b.c
index fcf05306..7e0bbf81 100644
--- a/drivers/gpu/nvgpu/gv11b/fifo_gv11b.c
+++ b/drivers/gpu/nvgpu/gv11b/fifo_gv11b.c
@@ -387,13 +387,12 @@ u32 gv11b_fifo_intr_0_error_mask(struct gk20a *g)
 
 u32 gv11b_fifo_get_preempt_timeout(struct gk20a *g)
 {
-        /* if timeouts are enabled, using 3000ms timeout
-         * for polling pdma/eng/runlist might kick in
-         * timeout handler in the cases where preempt
-         * is stuck. Use 1000ms timeout for polling when
-         * timeouts are enabled */
-        return nvgpu_is_timeouts_enabled(g) ? PREEMPT_TIMEOUT_1000_MS :
-                g->gr_idle_timeout_default;
+        /* using gr_idle_timeout for polling pdma/eng/runlist
+         * might kick in timeout handler in the cases where
+         * preempt is stuck. Use fifo_eng_timeout converted to ms
+         * for preempt polling */
+
+        return g->fifo_eng_timeout_us / 1000 ;
 }
 
 static int gv11b_fifo_poll_pbdma_chan_status(struct gk20a *g, u32 id,