diff --git a/drivers/gpu/nvgpu/include/nvgpu/gk20a.h b/drivers/gpu/nvgpu/include/nvgpu/gk20a.h
index 61da9ad76..68dd1018f 100644
--- a/drivers/gpu/nvgpu/include/nvgpu/gk20a.h
+++ b/drivers/gpu/nvgpu/include/nvgpu/gk20a.h
@@ -1376,8 +1376,6 @@ static inline u32 nvgpu_get_poll_timeout(struct gk20a *g)
 #define GK20A_NONSTALL_OPS_WAKEUP_SEMAPHORE	BIT32(0)
 #define GK20A_NONSTALL_OPS_POST_EVENTS		BIT32(1)
 
-bool is_nvgpu_gpu_state_valid(struct gk20a *g);
-
 /** IO Resource in the device tree for BAR0 */
 #define GK20A_BAR0_IORESOURCE_MEM	0U
 /** IO Resource in the device tree for BAR1 */
diff --git a/drivers/gpu/nvgpu/include/nvgpu/nvgpu_init.h b/drivers/gpu/nvgpu/include/nvgpu/nvgpu_init.h
index 74dbbdb0d..25d2fd796 100644
--- a/drivers/gpu/nvgpu/include/nvgpu/nvgpu_init.h
+++ b/drivers/gpu/nvgpu/include/nvgpu/nvgpu_init.h
@@ -313,4 +313,14 @@ int gk20a_busy(struct gk20a *g);
  */
 void gk20a_idle(struct gk20a *g);
 
+/**
+ * @brief Check if the GPU HW is in a valid state by making sure the boot_0
+ * register returns a valid value.
+ *
+ * @param g [in] The GPU
+ *
+ * @return True if the HW state is determined to be valid. False, otherwise.
+ */
+bool is_nvgpu_gpu_state_valid(struct gk20a *g);
+
 #endif /* NVGPU_INIT_H */