1 files changed, 2 insertions, 1 deletions

diff --git a/drivers/gpu/nvgpu/gp106/pmu_gp106.h b/drivers/gpu/nvgpu/gp106/pmu_gp106.h
index 3213b25c..5f399b89 100644
--- a/drivers/gpu/nvgpu/gp106/pmu_gp106.h
+++ b/drivers/gpu/nvgpu/gp106/pmu_gp106.h
@@ -18,8 +18,9 @@
         gk20a_dbg(gpu_dbg_pmu, fmt, ##arg)
 
 void gp106_init_pmu_ops(struct gpu_ops *gops);
-int gp106_pmu_reset(struct gk20a *g);
 void gp106_pmu_elpg_statistics(struct gk20a *g, u32 pg_engine_id,
                 struct pmu_pg_stats_data *pg_stat_data);
+bool gp106_pmu_is_engine_in_reset(struct gk20a *g);
+int gp106_pmu_engine_reset(struct gk20a *g, bool do_reset);
 
 #endif /*__PMU_GP106_H_*/