1 files changed, 2 insertions, 7 deletions

diff --git a/drivers/gpu/nvgpu/gp10b/gr_gp10b.c b/drivers/gpu/nvgpu/gp10b/gr_gp10b.c
index a6d72907..78be072f 100644
--- a/drivers/gpu/nvgpu/gp10b/gr_gp10b.c
+++ b/drivers/gpu/nvgpu/gp10b/gr_gp10b.c
@@ -49,8 +49,6 @@
 #include <nvgpu/hw/gp10b/hw_mc_gp10b.h>
 #include <nvgpu/hw/gp10b/hw_fuse_gp10b.h>
 
-#define NVGPU_GFXP_WFI_TIMEOUT_US       100LL
-
 bool gr_gp10b_is_valid_class(struct gk20a *g, u32 class_num)
 {
         bool valid = false;
@@ -2336,11 +2334,8 @@ int gp10b_gr_fuse_override(struct gk20a *g)
 int gr_gp10b_init_preemption_state(struct gk20a *g)
 {
         u32 debug_2;
-        u64 sysclk_rate;
-        u32 sysclk_cycles;
-
-        sysclk_rate = g->ops.clk.get_rate(g, CTRL_CLK_DOMAIN_GPCCLK);
-        sysclk_cycles = (u32)((sysclk_rate * NVGPU_GFXP_WFI_TIMEOUT_US) / 1000000ULL);
+        struct gr_gk20a *gr = &g->gr;
+        u32 sysclk_cycles = gr->gfxp_wfi_timeout_count;
         gk20a_writel(g, gr_fe_gfxp_wfi_timeout_r(),
                         gr_fe_gfxp_wfi_timeout_count_f(sysclk_cycles));