1 files changed, 2 insertions, 0 deletions

diff --git a/drivers/gpu/nvgpu/gk20a/gk20a_allocator_page.c b/drivers/gpu/nvgpu/gk20a/gk20a_allocator_page.c
index 521a87b3..07e3d823 100644
--- a/drivers/gpu/nvgpu/gk20a/gk20a_allocator_page.c
+++ b/drivers/gpu/nvgpu/gk20a/gk20a_allocator_page.c
@@ -168,6 +168,8 @@ static struct gk20a_page_alloc *__gk20a_alloc_pages(
                         }
                 } while (!chunk_addr && chunk_len >= a->page_size);
 
+                chunk_pages = chunk_len >> a->page_shift;
+
                 if (!chunk_addr) {
                         palloc_dbg(a, "bailing @ 0x%llx\n", chunk_len);
                         goto fail_cleanup;