1 files changed, 9 insertions, 0 deletions

diff --git a/drivers/gpu/nvgpu/common/pramin.c b/drivers/gpu/nvgpu/common/pramin.c
index ae9c9b1f..7955e6c2 100644
--- a/drivers/gpu/nvgpu/common/pramin.c
+++ b/drivers/gpu/nvgpu/common/pramin.c
@@ -16,6 +16,7 @@
 
 #include <nvgpu/pramin.h>
 #include <nvgpu/page_allocator.h>
+#include <nvgpu/enabled.h>
 
 #include "gk20a/gk20a.h"
 
@@ -88,6 +89,14 @@ void nvgpu_pramin_access_batched(struct gk20a *g, struct nvgpu_mem *mem,
         void *sgl;
         u32 byteoff, start_reg, until_end, n;
 
+        /*
+         * TODO: Vidmem is not accesible through pramin on shutdown path.
+         * driver should be refactored to prevent this from happening, but for
+         * now it is ok just to ignore the writes
+         */
+        if (!g->regs && nvgpu_is_enabled(g, NVGPU_DRIVER_IS_DYING))
+                return;
+
         alloc = get_vidmem_page_alloc(mem->priv.sgt->sgl);
         sgt = &alloc->sgt;
         for (sgl = sgt->sgl; sgl; sgl = nvgpu_sgt_get_next(sgt, sgl)) {