1 files changed, 10 insertions, 2 deletions

diff --git a/drivers/gpu/nvgpu/gk20a/mm_gk20a.c b/drivers/gpu/nvgpu/gk20a/mm_gk20a.c
index 4005d5f4..1b052a20 100644
--- a/drivers/gpu/nvgpu/gk20a/mm_gk20a.c
+++ b/drivers/gpu/nvgpu/gk20a/mm_gk20a.c
@@ -3203,8 +3203,16 @@ int gk20a_gmmu_alloc_map(struct vm_gk20a *vm, size_t size,
 int gk20a_gmmu_alloc_map_attr(struct vm_gk20a *vm,
                          enum dma_attr attr, size_t size, struct mem_desc *mem)
 {
-        if (vm->mm->vidmem_is_vidmem)
-                return gk20a_gmmu_alloc_map_attr_vid(vm, 0, size, mem);
+        if (vm->mm->vidmem_is_vidmem) {
+                int err = gk20a_gmmu_alloc_map_attr_vid(vm, 0, size, mem);
+
+                if (!err)
+                        return 0;
+                /*
+                 * Fall back to sysmem (which may then also fail) in case
+                 * vidmem is exhausted.
+                 */
+        }
 
         return gk20a_gmmu_alloc_map_attr_sys(vm, 0, size, mem);
 }