1 files changed, 5 insertions, 6 deletions

diff --git a/drivers/gpu/drm/amd/amdgpu/gmc_v9_0.c b/drivers/gpu/drm/amd/amdgpu/gmc_v9_0.c
index a29a31f5b012..787d20b5bbb9 100644
--- a/drivers/gpu/drm/amd/amdgpu/gmc_v9_0.c
+++ b/drivers/gpu/drm/amd/amdgpu/gmc_v9_0.c
@@ -415,6 +415,11 @@ static void gmc_v9_0_vram_gtt_location(struct amdgpu_device *adev,
         amdgpu_vram_location(adev, &adev->mc, base);
         adev->mc.gtt_base_align = 0;
         amdgpu_gtt_location(adev, mc);
+        /* base offset of vram pages */
+        if (adev->flags & AMD_IS_APU)
+                adev->vm_manager.vram_base_offset = gfxhub_v1_0_get_mc_fb_offset(adev);
+        else
+                adev->vm_manager.vram_base_offset = 0;
 }
 
 /**
@@ -547,12 +552,6 @@ static int gmc_v9_0_vm_init(struct amdgpu_device *adev)
                 adev->vm_manager.num_level = 3;
         amdgpu_vm_manager_init(adev);
 
-        /* base offset of vram pages */
-        if (adev->flags & AMD_IS_APU)
-                adev->vm_manager.vram_base_offset = gfxhub_v1_0_get_mc_fb_offset(adev);
-        else
-                adev->vm_manager.vram_base_offset = 0;
-
         return 0;
 }