2 files changed, 3 insertions, 5 deletions

diff --git a/drivers/gpu/nvgpu/gk20a/mm_gk20a.c b/drivers/gpu/nvgpu/gk20a/mm_gk20a.c
index 13dc4241..8ac6d1d1 100644
--- a/drivers/gpu/nvgpu/gk20a/mm_gk20a.c
+++ b/drivers/gpu/nvgpu/gk20a/mm_gk20a.c
@@ -914,13 +914,13 @@ static int gk20a_init_vidmem(struct mm_gk20a *mm)
         int err;
 
         static struct gk20a_alloc_carveout wpr_co =
-                GK20A_CARVEOUT("wpr-region",
-                               NVGPU_VIDMEM_BOOTSTRAP_ALLOCATOR_BASE, SZ_16M);
+                GK20A_CARVEOUT("wpr-region", 0, SZ_16M);
 
         if (!size)
                 return 0;
 
-        bootstrap_base = NVGPU_VIDMEM_BOOTSTRAP_ALLOCATOR_BASE;
+        wpr_co.base = size - SZ_256M;
+        bootstrap_base = wpr_co.base;
         bootstrap_size = SZ_16M;
         base = default_page_size;
 
diff --git a/drivers/gpu/nvgpu/gk20a/mm_gk20a.h b/drivers/gpu/nvgpu/gk20a/mm_gk20a.h
index 36ba6248..e1ee8c2e 100644
--- a/drivers/gpu/nvgpu/gk20a/mm_gk20a.h
+++ b/drivers/gpu/nvgpu/gk20a/mm_gk20a.h
@@ -51,8 +51,6 @@ enum gk20a_aperture {
         APERTURE_VIDMEM
 };
 
-#define NVGPU_VIDMEM_BOOTSTRAP_ALLOCATOR_BASE   0x18000000
-
 static inline const char *gk20a_aperture_str(enum gk20a_aperture aperture)
 {
         switch (aperture) {