diff --git a/drivers/gpu/nvgpu/common/gr/gr.c b/drivers/gpu/nvgpu/common/gr/gr.c
index c570572ec..9d37fbe5b 100644
--- a/drivers/gpu/nvgpu/common/gr/gr.c
+++ b/drivers/gpu/nvgpu/common/gr/gr.c
@@ -620,6 +620,8 @@ static int gr_reset_engine(struct gk20a *g)
 		nvgpu_device_get(g, NVGPU_DEVTYPE_GRAPHICS,
 			nvgpu_gr_get_syspipe_id(g, g->mig.cur_gr_instance));
 
+	nvgpu_assert(dev != NULL);
+
 	nvgpu_log(g, gpu_dbg_gr, "Reset GR%u", cur_gr_instance_id);
 
 	/* Reset GR engine: Disable then enable GR engine */
diff --git a/drivers/gpu/nvgpu/hal/grmgr/grmgr_ga10b.c b/drivers/gpu/nvgpu/hal/grmgr/grmgr_ga10b.c
index adff3c11e..18414f3a2 100644
--- a/drivers/gpu/nvgpu/hal/grmgr/grmgr_ga10b.c
+++ b/drivers/gpu/nvgpu/hal/grmgr/grmgr_ga10b.c
@@ -803,6 +803,8 @@ int ga10b_grmgr_init_gr_manager(struct gk20a *g)
 		gr_dev = nvgpu_device_get(g, NVGPU_DEVTYPE_GRAPHICS,
 			gr_syspipe->gr_syspipe_id);
 
+		nvgpu_assert(gr_dev != NULL);
+
 		/*
 		 * HW recommended to put GR engine into reset before programming
 		 * config_gpc_rs_map (ga10b_grmgr_config_gpc_smc_map()).